[ET][EZ] Print tensor data for Half dtype

jorgep31415 · jorgep31415 · commit b303b3c5cb5f · 2024-05-30T11:13:17.000-07:00
Useful for debugging `Half` i.e. `fp16` models. Differential Revision: [D57977366](https://our.internmc.facebook.com/intern/diff/D57977366/) [ghstack-poisoned]
diff --git a/extension/evalue_util/print_evalue.cpp b/extension/evalue_util/print_evalue.cpp
@@ -158,7 +158,7 @@ void print_tensor(std::ostream& os, exec_aten::Tensor tensor) {
     break;
 
   switch (tensor.scalar_type()) {
-    ET_FORALL_REAL_TYPES_AND(Bool, PRINT_TENSOR_DATA)
+    ET_FORALL_REAL_TYPES_AND2(Bool, Half, PRINT_TENSOR_DATA)
     default:
       os << "[<unhandled scalar type " << (int)tensor.scalar_type() << ">]";
   }

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