print bfloat16 tensor data (pytorch#14889)

manuelcandales · web-flow · commit 38b51aa7e5b6 · 2025-10-08T21:39:40.000-04:00
Handle bfloat16 when printing tensor data.
diff --git a/extension/evalue_util/print_evalue.cpp b/extension/evalue_util/print_evalue.cpp
@@ -160,7 +160,7 @@ void print_tensor(std::ostream& os, executorch::aten::Tensor tensor) {
     break;
 
   switch (tensor.scalar_type()) {
-    ET_FORALL_REAL_TYPES_AND2(Bool, Half, PRINT_TENSOR_DATA)
+    ET_FORALL_REALHBBF16_TYPES(PRINT_TENSOR_DATA)
     default:
       os << "[<unhandled scalar type " << (int)tensor.scalar_type() << ">]";
   }

Original file line number	Diff line number	Diff line change
`@@ -160,7 +160,7 @@ void print_tensor(std::ostream& os, executorch::aten::Tensor tensor) {`
`160`	`160`	`break;`
`161`	`161`
`162`	`162`	`switch (tensor.scalar_type()) {`
`163`		`- ET_FORALL_REAL_TYPES_AND2(Bool, Half, PRINT_TENSOR_DATA)`
	`163`	`+ ET_FORALL_REALHBBF16_TYPES(PRINT_TENSOR_DATA)`
`164`	`164`	`default:`
`165`	`165`	`os << "[<unhandled scalar type " << (int)tensor.scalar_type() << ">]";`
`166`	`166`	`}`