__init__.py 753 B

12345678910111213141516171819
  1. from .lazy_ir import (
  2. generate_non_native_lazy_ir_nodes as generate_non_native_lazy_ir_nodes,
  3. GenLazyIR as GenLazyIR,
  4. GenLazyNativeFuncDefinition as GenLazyNativeFuncDefinition,
  5. GenLazyShapeInferenceDefinition as GenLazyShapeInferenceDefinition,
  6. )
  7. from .native_functions import (
  8. compute_native_function_declaration as compute_native_function_declaration,
  9. )
  10. from .register_dispatch_key import (
  11. gen_registration_headers as gen_registration_headers,
  12. gen_registration_helpers as gen_registration_helpers,
  13. RegisterDispatchKey as RegisterDispatchKey,
  14. )
  15. from .ufunc import (
  16. compute_ufunc_cpu as compute_ufunc_cpu,
  17. compute_ufunc_cpu_kernel as compute_ufunc_cpu_kernel,
  18. compute_ufunc_cuda as compute_ufunc_cuda,
  19. )