From bc9c861235e23d4052acc80f456252287d5b2213 Mon Sep 17 00:00:00 2001
From: Henry Hu <hhh@meta.com>
Date: Fri, 7 Jul 2023 16:35:16 -0700
Subject: [PATCH] Add converter for acc_op.dtype (#816)

Summary:
Pull Request resolved: https://github.com/facebookincubator/AITemplate/pull/816

Support .type() op.

Differential Revision: D47283348

fbshipit-source-id: fa853d3ee02d6a5a0e67ea3ef428567c07706419
---
 fx2ait/fx2ait/converters/ait_converters.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/fx2ait/fx2ait/converters/ait_converters.py b/fx2ait/fx2ait/converters/ait_converters.py
index 88f7c8e34..ab4418987 100644
--- 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, ...],