Encode is_cpu_op as a boolean value in the converter

mlcommons · Jul 1, 2024 · 8109eac · 8109eac
1 parent b8b2b74
commit 8109eac
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