已合并
fix_generate_extern_kernel_alloc #37537
cuiduo创建于 6月3日
fix_generate_extern_kernel_alloc #37537
已合并
共 2 个文件变更+1-2
| @@ -8,7 +8,6 @@ import torch_npu._inductor | |||
| 8 | 8 | ||
| 9 | class TestAdd(TestUtils): | 9 | class TestAdd(TestUtils): |
| 10 | 10 | ||
| 11 | - | ||
| 12 | def op_calc(self, first_element, second_element): | 11 | def op_calc(self, first_element, second_element): |
| 13 | result = first_element + second_element | 12 | result = first_element + second_element |
| 14 | return result | 13 | return result |
| @@ -93,7 +93,7 @@ class NpuMlirWrapperCodeGen(PythonWrapperCodegen): | |||
| 93 | """ | 93 | """ |
| 94 | ) | 94 | ) |
| 95 | 95 | ||
| 96 | - def generate_extern_kernel_alloc(self, extern_kernel, args): | 96 | + def _generate_extern_kernel_alloc_helper(self, extern_kernel, args): |
| 97 | # If it's a NoneLayout then the extern_kernel should essentially be | 97 | # If it's a NoneLayout then the extern_kernel should essentially be |
| 98 | # treated as if it doesn't return anything | 98 | # treated as if it doesn't return anything |
| 99 | no_return = isinstance(extern_kernel.layout, ir.NoneLayout) | 99 | no_return = isinstance(extern_kernel.layout, ir.NoneLayout) |