Add converter for acc_op.dtype

Differential Revision: D47283348 fbshipit-source-id: 9b6298e0cb831afb06bc1456d50ddd92205810dc
henryhu6 · Jul 8, 2023 · 8e3d54d · 8e3d54d
1 parent 8653901
commit 8e3d54d
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, ...],