mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
a4647cc1fa
Previous batches: * https://github.com/pytorch/pytorch/pull/81285 * https://github.com/pytorch/pytorch/pull/81335 We have multiple batches here to minimize merge conflicts and reviewing process. Once everything has been formatted by ufmt (black+usort), the current black linter will be removed. Pull Request resolved: https://github.com/pytorch/pytorch/pull/81570 Approved by: https://github.com/ezyang
20 lines
753 B
Python
20 lines
753 B
Python
from .lazy_ir import (
|
|
generate_non_native_lazy_ir_nodes as generate_non_native_lazy_ir_nodes,
|
|
GenLazyIR as GenLazyIR,
|
|
GenLazyNativeFuncDefinition as GenLazyNativeFuncDefinition,
|
|
GenLazyShapeInferenceDefinition as GenLazyShapeInferenceDefinition,
|
|
)
|
|
from .native_functions import (
|
|
compute_native_function_declaration as compute_native_function_declaration,
|
|
)
|
|
from .register_dispatch_key import (
|
|
gen_registration_headers as gen_registration_headers,
|
|
gen_registration_helpers as gen_registration_helpers,
|
|
RegisterDispatchKey as RegisterDispatchKey,
|
|
)
|
|
from .ufunc import (
|
|
compute_ufunc_cpu as compute_ufunc_cpu,
|
|
compute_ufunc_cpu_kernel as compute_ufunc_cpu_kernel,
|
|
compute_ufunc_cuda as compute_ufunc_cuda,
|
|
)
|