diff --git a/codegen/autograd/gen_variable_type.py b/codegen/autograd/gen_variable_type.py index a6d45c2c4179499da97ae0de269e915b70332811..3fda4bfc54ff98c20ec0673c9851336e8dd07fe9 100644 --- a/codegen/autograd/gen_variable_type.py +++ b/codegen/autograd/gen_variable_type.py @@ -122,6 +122,7 @@ def gen_variable_type_func( type_definition = re.sub(r'at::redispatch::(\w+)_outf', r'at_npu::redispatch::\1_out', type_definition) else: type_definition = type_definition.replace('at::redispatch', 'at_npu::redispatch') + type_definition = type_definition.replace('_symint', '') wrapper_registration = gen_wrapper_registration(f, "Default") result[f"type_derived_method_definitions"] = [type_definition]