add print in AsBlasDataType

openxla · May 23, 2023 · 7e9f1d6 · 7e9f1d6
1 parent d836d4c
commit 7e9f1d6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/xla/service/gpu/matmul_utils.cc b/xla/service/gpu/matmul_utils.cc
@@ -738,7 +738,8 @@ StatusOr<se::blas::DataType> AsBlasDataType(PrimitiveType dtype) {
     case C128:
       return se::blas::DataType::kComplexDouble;
     default:
-      return InternalError("AsBlasDataType: unsupported type");
+      return InternalError("AsBlasDataType: unsupported type: %s",
+                           primitive_util::LowercasePrimitiveTypeName(dtype));
   }
 }