tilelang.jit.adapter package

Subpackages

• tilelang.jit.adapter.ctypes package
  • Submodules
    • tilelang.jit.adapter.ctypes.adapter module
      • CtypesKernelAdapter
  • Module contents
• tilelang.jit.adapter.cython package
  • Submodules
    • tilelang.jit.adapter.cython.adapter module
      • CythonKernelAdapter
      • get_cache_dir()
      • get_cached_lib()
      • get_cython_compiler()
    • tilelang.jit.adapter.cython.cython_wrapper module
      • CythonKernelWrapper
  • Module contents

Submodules

• tilelang.jit.adapter.base module
  • BaseKernelAdapter
    • BaseKernelAdapter.func
    • BaseKernelAdapter.get_kernel_source()
• tilelang.jit.adapter.dlpack module
  • TorchDLPackKernelAdapter
• tilelang.jit.adapter.libgen module
  • LibraryGenerator
    • LibraryGenerator.compile_lib()
    • LibraryGenerator.get_lib_path()
    • LibraryGenerator.get_source_path()
    • LibraryGenerator.lib_code
    • LibraryGenerator.libpath
    • LibraryGenerator.load_lib()
    • LibraryGenerator.remove_lib()
    • LibraryGenerator.set_lib_path()
    • LibraryGenerator.set_src_path()
    • LibraryGenerator.srcpath
    • LibraryGenerator.update_lib_code()
• tilelang.jit.adapter.utils module
  • get_annotated_mod()
  • is_cpu_target()
  • is_cuda_target()
  • is_hip_target()
  • match_declare_kernel()
  • match_declare_kernel_cpu()
  • match_global_kernel()
• tilelang.jit.adapter.wrapper module
  • BaseWrapper
    • BaseWrapper.wrap()
  • TLCPUSourceWrapper
    • TLCPUSourceWrapper.CALL_PREFIX
    • TLCPUSourceWrapper.INIT_FUNC
    • TLCPUSourceWrapper.backend
    • TLCPUSourceWrapper.create_call_func()
    • TLCPUSourceWrapper.device_mod
    • TLCPUSourceWrapper.get_cpu_init_func()
    • TLCPUSourceWrapper.get_dynamic_symbolic_set()
    • TLCPUSourceWrapper.host_mod
    • TLCPUSourceWrapper.parse_source_information()
    • TLCPUSourceWrapper.pass_configs
    • TLCPUSourceWrapper.prim_func
    • TLCPUSourceWrapper.update_lib_code()
  • TLCUDASourceWrapper
    • TLCUDASourceWrapper.backend
    • TLCUDASourceWrapper.create_dispatch_func()
    • TLCUDASourceWrapper.device_mod
    • TLCUDASourceWrapper.generate_tma_descriptor_args()
    • TLCUDASourceWrapper.get_cuda_init_func()
    • TLCUDASourceWrapper.get_dynamic_symbolic_set()
    • TLCUDASourceWrapper.host_mod
    • TLCUDASourceWrapper.is_tma_descriptor_arg()
    • TLCUDASourceWrapper.parse_source_information()
    • TLCUDASourceWrapper.pass_configs
    • TLCUDASourceWrapper.prim_func
    • TLCUDASourceWrapper.update_lib_code()
  • TLHIPSourceWrapper
    • TLHIPSourceWrapper.get_hip_init_func()
    • TLHIPSourceWrapper.get_stream_type()
  • TLWrapper
    • TLWrapper.assign_device_module()
    • TLWrapper.assign_host_module()
    • TLWrapper.assign_optimized_module()
    • TLWrapper.assign_pass_configs()
    • TLWrapper.device_mod
    • TLWrapper.host_mod
    • TLWrapper.lib
    • TLWrapper.pass_configs
    • TLWrapper.target
    • TLWrapper.wrap()

Module contents