use get_const_proxy to construct const op

DeepLink-org · Jan 17, 2024 · daccc34 · daccc34
1 parent fb2cfc7
commit daccc34
Show file tree

Hide file tree

Showing 2 changed files with 90 additions and 114 deletions.
diff --git a/dicp/dicp/vendor/AscendGraph/codegen/utils.py b/dicp/dicp/vendor/AscendGraph/codegen/utils.py
@@ -36,7 +36,7 @@ def get_ascend_dtype(dtype: torch.dtype) -> str:
         return "BOOL"
     elif dtype == torch.int64:
         return "INT64"
-    elif dtype == torch.float32:
+    elif dtype in [torch.float32, torch.float]:
         return "FLOAT"
     elif dtype == torch.float16:
         return "FLOAT16"