已合并
[feat] exposed customed api #31388
culechan创建于 3月4日
[feat] exposed customed api #31388
已合并
共 3 个文件变更+3-1
| @@ -23,7 +23,7 @@ ParsedYaml = namedtuple('ParsedYaml', ['native_functions', 'backend_indices']) | |||
| 23 | 23 | ||
| 24 | 24 | ||
| 25 | CUSTOM_FUNCTIONS_DECLARATION = CodeTemplate("""\ | 25 | CUSTOM_FUNCTIONS_DECLARATION = CodeTemplate("""\ |
| 26 | -${return_type} ${func_name}(${args_str}); | 26 | +TORCH_NPU_API ${return_type} ${func_name}(${args_str}); |
| 27 | """) | 27 | """) |
| 28 | 28 | ||
| 29 | CUSTOM_FUNCTIONS_DEFINITION = CodeTemplate("""\ | 29 | CUSTOM_FUNCTIONS_DEFINITION = CodeTemplate("""\ |
| @@ -2,6 +2,7 @@ | |||
| 2 | 2 | ||
| 3 | 3 | ||
| 4 | 4 | ||
| 5 | + | ||
| 5 | 6 | ||
| 6 | namespace at_npu { | 7 | namespace at_npu { |
| 7 | namespace native { | 8 | namespace native { |
| @@ -2,6 +2,7 @@ | |||
| 2 | 2 | ||
| 3 | 3 | ||
| 4 | 4 | ||
| 5 | + | ||
| 5 | 6 | ||
| 6 | namespace at_npu { | 7 | namespace at_npu { |
| 7 | namespace redispatch { | 8 | namespace redispatch { |