Add converter for acc_op.dtype

Differential Revision: D47283348 fbshipit-source-id: 601c3833ec95f0077d4b6b6ad34aba96f715be12
facebookincubator · Jul 7, 2023 · 125cf41 · 125cf41
1 parent 8653901
commit 125cf41
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, ...],