Menooker commented on a change in pull request #5601:
URL: https://github.com/apache/incubator-tvm/pull/5601#discussion_r428427929
##########
File path: include/tvm/runtime/data_type.h
##########
@@ -81,6 +82,10 @@ class DataType {
bool is_float() const { return code() == DataType::kFloat; }
/*! \return whether type is a float16 type. */
bool is_float16() const { return is_float() && bits() == 16; }
+ /*! \return whether type is a bfloat type. */
+ bool is_bfloat() const { return code() == DataType::kBFloat; }
Review comment:
ok, changed
##########
File path: include/tvm/runtime/data_type.h
##########
@@ -297,6 +302,8 @@ inline const char* TypeCode2Str(int type_code) {
return "Object";
case kTVMObjectRValueRefArg:
return "ObjectRValueRefArg";
+ case kTVMBFloat:
+ return "bf";
Review comment:
ok, changed
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
