diff --git a/codegen/autograd/gen_variable_type.py b/codegen/autograd/gen_variable_type.py index 6649f46dd2e09e0d9fcd4fe813abb47c62f1968d..5afd25eccad8e1588ae2c6d3c541080a4ebf0ebe 100644 --- a/codegen/autograd/gen_variable_type.py +++ b/codegen/autograd/gen_variable_type.py @@ -123,6 +123,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]