mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
torchgen: move dispatch_helpers out of RegisterDispatchDefinitions.ini (#144363)
The dispatch_helpers should be generated once, not once per kernel namespace. Differential Revision: [D67925497](https://our.internmc.facebook.com/intern/diff/D67925497/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/144363 Approved by: https://github.com/bdhirsh
This commit is contained in:
parent
7a93a58b3c
commit
603e1c0b02
4 changed files with 10 additions and 7 deletions
|
|
@ -5,8 +5,6 @@ ${ns_prologue}
|
|||
// at namespace already.
|
||||
namespace {
|
||||
|
||||
${dispatch_helpers}
|
||||
|
||||
${dispatch_anonymous_definitions}
|
||||
|
||||
${static_init_dispatch_registrations}
|
||||
|
|
|
|||
|
|
@ -51,5 +51,11 @@ $external_backend_headers
|
|||
$dispatch_headers
|
||||
$ops_headers
|
||||
|
||||
namespace at {
|
||||
namespace {
|
||||
$dispatch_helpers
|
||||
} // namespace
|
||||
} // namespace at
|
||||
|
||||
// See template file RegisterDispatchDefinitions.ini
|
||||
$dispatch_definitions
|
||||
|
|
|
|||
|
|
@ -1642,9 +1642,6 @@ TORCH_LIBRARY_IMPL({namespace}, {dispatch_key}, m) {{
|
|||
lambda: {
|
||||
"ns_prologue": ns_helper.prologue,
|
||||
"ns_epilogue": ns_helper.epilogue,
|
||||
"dispatch_helpers": dest.gen_registration_helpers(backend_idx)
|
||||
if gen_dispatch_helpers
|
||||
else [],
|
||||
"dispatch_anonymous_definitions": anonymous_definitions[
|
||||
kernel_namespace
|
||||
],
|
||||
|
|
@ -2331,7 +2328,9 @@ def gen_source_files(
|
|||
backend_index, per_operator_headers, rocm
|
||||
),
|
||||
"ops_headers": operator_headers(),
|
||||
"dispatch_helpers": "",
|
||||
"dispatch_helpers": dest.gen_registration_helpers(backend_index)
|
||||
if gen_dispatch_helpers
|
||||
else [],
|
||||
"dispatch_definitions": dispatch_definitions,
|
||||
},
|
||||
)
|
||||
|
|
|
|||
|
|
@ -499,6 +499,7 @@ TORCH_API void Register${backend_name}${dispatch_key}NativeFunctions() {
|
|||
"dispatch_headers": dest.gen_registration_headers(
|
||||
backend_index, per_operator_headers=per_operator_headers, rocm=False
|
||||
),
|
||||
"dispatch_helpers": dest.gen_registration_helpers(backend_index),
|
||||
"dispatch_definitions": fm.substitute_with_template(
|
||||
"RegisterDispatchDefinitions.ini",
|
||||
lambda: {
|
||||
|
|
@ -506,7 +507,6 @@ TORCH_API void Register${backend_name}${dispatch_key}NativeFunctions() {
|
|||
"ns_epilogue": ns_helper.epilogue,
|
||||
"static_init_dispatch_registrations": static_init_dispatch_registrations,
|
||||
"deferred_dispatch_registrations": deferred_dispatch_registrations,
|
||||
"dispatch_helpers": dest.gen_registration_helpers(backend_index),
|
||||
"dispatch_namespace": dispatch_key.lower(),
|
||||
"dispatch_namespaced_definitions": "",
|
||||
"dispatch_anonymous_definitions": list(
|
||||
|
|
|
|||
Loading…
Reference in a new issue