Add converter for acc_op.dtype

Differential Revision: D47283348 fbshipit-source-id: 6ae0509e593ca2fbb59bf89507dfeff0816e777b
facebookincubator · Jul 8, 2023 · 34de7fb · 34de7fb
1 parent 8653901
commit 34de7fb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/fx2ait/fx2ait/converters/ait_converters.py b/fx2ait/fx2ait/converters/ait_converters.py
@@ -1550,6 +1550,7 @@ def acc_ops_contiguous(
 
 
 @ait_converter(acc_ops.to_dtype)
+@ait_converter(acc_ops.dtype)
 def acc_ops_to_dtype(
     target: Target,
     args: Tuple[Argument, ...],