@@ -43,9 +43,11 @@ using cutlass::bfloat16_t;
|
||||
// Display utilities
|
||||
//
|
||||
|
||||
#if !defined(__CUDACC_RTC__)
|
||||
CUTE_HOST std::ostream& operator<<(std::ostream& os, bfloat16_t const& v)
|
||||
{
|
||||
return os << float(v);
|
||||
}
|
||||
#endif
|
||||
|
||||
} // end namespace cute
|
||||
|
||||
Reference in New Issue
Block a user