Merge pull request #114 from mlcommons/feeder-bug-fix

Encode is_cpu_op as a boolean value in the converter
mlcommons · Jul 2, 2024 · 5143d92 · 5143d92
2 parents c0643cd + 6e98660
commit 5143d92
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/converter/pytorch_converter.py b/src/converter/pytorch_converter.py
@@ -337,7 +337,7 @@ def convert_to_chakra_node(
                 ChakraAttr(name="tid", int64_val=pytorch_node.tid),
                 ChakraAttr(name="fw_tid", int64_val=pytorch_node.fw_tid),
                 ChakraAttr(name="op_schema", string_val=pytorch_node.op_schema),
-                ChakraAttr(name="is_cpu_op", int32_val=not pytorch_node.is_gpu_op()),
+                ChakraAttr(name="is_cpu_op", bool_val=not pytorch_node.is_gpu_op()),
             ]
         )
         return chakra_node