From 347ea4e2212436672adcf1c8f64c5bdc707193f8 Mon Sep 17 00:00:00 2001 From: wgb Date: Tue, 26 Aug 2025 17:16:24 +0800 Subject: [PATCH] npu codegen autograd support symint --- codegen/autograd/gen_variable_type.py | 1 + 1 file changed, 1 insertion(+) diff --git a/codegen/autograd/gen_variable_type.py b/codegen/autograd/gen_variable_type.py index a6d45c2c417..3fda4bfc54f 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] -- Gitee