Rename format_cast function name.

DeepLink-org · Jan 17, 2024 · 6e71ed4 · 6e71ed4
1 parent 016198a
commit 6e71ed4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dipu/torch_dipu/dipu/utils.py b/dipu/torch_dipu/dipu/utils.py
@@ -51,7 +51,7 @@ def wrap(x):
         return tree_map(wrap, out)
 
 
-def format_cast(tensor:torch.Tensor, format:_C.NativeMemoryFormat) -> torch.Tensor:
+def ascend_format_cast(tensor:torch.Tensor, format:_C.NativeMemoryFormat) -> torch.Tensor:
     if isinstance(tensor, torch.Tensor) and 'FRACTAL_NZ' in str(format):
         return AscendFormatTensor(torch_dipu.native_memory_format_cast(tensor, format))
     if isinstance(tensor, AscendFormatTensor) and not 'FRACTAL_NZ' in str(format):