From 420cc109d253437f999482156c9ff5b21f464051 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 6649f46dd2..5afd25ecca 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] -- Gitee