ejkernel.callib package

• ejkernel.callib._ejit
  • ejit()
  • get_hash_of_lowering()
  • get_safe_hash_int()
  • hash_fn()
  • load_cached_functions()
  • load_compiled_fn()
  • save_compiled_fn()
  • smart_compile()
• ejkernel.callib._pallas_call
  • buffered_pallas_call()
• ejkernel.callib._triton_call
  • CompilationResult
  • aval_size_bytes()
  • avals_to_layouts()
  • compile_ttir_inplace()
  • compile_ttir_to_hsaco_inplace()
  • compile_ttir_to_ptx_inplace()
  • get_cuda_backend()
  • get_hip_backend()
  • get_or_create_triton_kernel()
  • get_triton_type()
  • normalize_grid()
  • triton_call()
  • triton_kernel_call_abstract_eval()
  • triton_kernel_call_lowering()
  • triton_kernel_call_raise_on_jvp()
  • triton_kernel_call_raise_on_vmap()
• ejkernel.callib._utils
  • DummyStream
  • ShapeDtype
  • cdiv()
  • check_bool_flag()
  • get_cache_dir()
  • next_power_of_2()
  • quiet()
  • strides_from_shape()