tilelang package

Subpackages

• tilelang.autotuner package
  • Submodules
    • tilelang.autotuner.param module
  • Module contents
• tilelang.cache package
  • Submodules
    • tilelang.cache.kernel_cache module
    • tilelang.cache.tuner_cache module
  • Module contents
• tilelang.carver package
  • Subpackages
    • tilelang.carver.arch package
      • Subpackages
      • Submodules
      • Module contents
    • tilelang.carver.roller package
      • Subpackages
      • Submodules
      • Module contents
    • tilelang.carver.template package
      • Submodules
      • Module contents
  • Submodules
    • tilelang.carver.analysis module
    • tilelang.carver.common_schedules module
    • tilelang.carver.matmul_analysis module
    • tilelang.carver.utils module
  • Module contents
• tilelang.common package
  • Submodules
    • tilelang.common.transform_kind module
  • Module contents
• tilelang.contrib package
  • Submodules
    • tilelang.contrib.cc module
      • create_executable()
      • create_shared()
      • create_staticlib()
      • cross_compiler()
      • get_cc()
      • get_cplus_compiler()
      • get_global_symbol_section_map()
      • get_target_by_dump_machine()
    • tilelang.contrib.dlpack module
      • convert_func()
      • to_pytorch_func()
    • tilelang.contrib.hipcc module
      • compile_hip()
    • tilelang.contrib.nvcc module
      • callback_libdevice_path()
      • compile_cuda()
      • find_cuda_path()
      • get_cuda_version()
      • get_nvcc_compiler()
      • have_cudagraph()
      • have_fp16()
      • have_int8()
      • have_tensorcore()
      • parse_compute_version()
    • tilelang.contrib.nvrtc module
    • tilelang.contrib.rocm module
      • find_lld()
      • find_rocm_path()
      • have_matrixcore()
      • parse_compute_version()
      • rocm_link()
  • Module contents
• tilelang.engine package
  • Submodules
    • tilelang.engine.callback module
      • register_cuda_postproc()
      • register_cuda_postproc_callback()
      • register_hip_postproc()
      • register_hip_postproc_callback()
    • tilelang.engine.lower module
      • canon_target_host()
      • device_codegen()
      • device_codegen_without_compile()
      • extrac_params()
      • get_device_call()
      • get_host_call()
      • has_device_kernel_launch()
      • host_codegen()
      • is_cpu_device_backend()
      • is_device_call()
      • is_device_call_c_device()
      • lower()
    • tilelang.engine.param module
      • CompiledArtifact
      • KernelParam
    • tilelang.engine.phase module
      • LowerAndLegalize()
      • OptimizeForTarget()
      • allow_fence_proxy()
      • allow_global_thread_synchronization()
      • allow_tma_and_warp_specialized()
      • allow_vectorize()
      • allow_warp_specialized()
  • Module contents
• tilelang.intrinsics package
  • Submodules
    • tilelang.intrinsics.mfma_layout module
    • tilelang.intrinsics.mfma_macro_generator module
    • tilelang.intrinsics.mma_layout module
    • tilelang.intrinsics.mma_macro_generator module
    • tilelang.intrinsics.utils module
  • Module contents
• tilelang.jit package
  • Subpackages
    • tilelang.jit.adapter package
      • Subpackages
      • Submodules
      • Module contents
  • Submodules
    • tilelang.jit.env module
    • tilelang.jit.kernel module
    • tilelang.jit.param module
  • Module contents
• tilelang.language package
  • Subpackages
    • tilelang.language.tir package
      • Submodules
      • Module contents
  • Submodules
    • tilelang.language.allocate module
    • tilelang.language.builtin module
    • tilelang.language.copy module
    • tilelang.language.customize module
    • tilelang.language.fill module
    • tilelang.language.frame module
    • tilelang.language.gemm module
    • tilelang.language.kernel module
    • tilelang.language.logical module
    • tilelang.language.memscope module
    • tilelang.language.parallel module
    • tilelang.language.persistent module
    • tilelang.language.pipeline module
    • tilelang.language.print module
    • tilelang.language.proxy module
    • tilelang.language.reduce module
    • tilelang.language.warpgroup module
  • Module contents
• tilelang.layout package
  • Submodules
    • tilelang.layout.fragment module
    • tilelang.layout.layout module
    • tilelang.layout.swizzle module
  • Module contents
• tilelang.math package
  • Module contents
• tilelang.primitives package
  • Subpackages
    • tilelang.primitives.gemm package
      • Submodules
      • Module contents
  • Module contents
• tilelang.profiler package
  • Submodules
    • tilelang.profiler.bench module
      • do_bench()
  • Module contents
• tilelang.quantize package
  • Submodules
    • tilelang.quantize.lop3 module
    • tilelang.quantize.quantization module
    • tilelang.quantize.utils module
  • Module contents
• tilelang.testing package
  • Module contents
• tilelang.tools package
  • Submodules
    • tilelang.tools.Analyzer module
    • tilelang.tools.plot_layout module
  • Module contents
• tilelang.transform package
  • Submodules
    • tilelang.transform.pass_config module
      • PassConfigKey
    • tilelang.transform.simplify module
      • Simplify()
      • apply_simplify()
      • simplify_prim_func()
  • Module contents
    • AlignDynamicSharedMemoryAllocations()
    • AnnotateDeviceRegions()
    • ClusterPlanning()
    • ConfigIndexBitwidth()
    • EliminateStorageSyncForMBarrier()
    • FlattenBuffer()
    • FrontendLegalize()
    • IfStmtBinding()
    • InjectFenceProxy()
    • InjectPTXAsyncCopy()
    • InjectSoftwarePipeline()
    • InjectTmaBarrier()
    • LayoutInference()
    • LegalizeSafeMemoryAccess()
    • LegalizeVectorizedLoop()
    • LoopVectorizeDynamic()
    • LowerCpengineIntrin()
    • LowerDeviceStorageAccessInfo()
    • LowerHopperIntrin()
    • LowerL2Persistent()
    • LowerTileOp()
    • MakePackedAPI()
    • MergeIfStmt()
    • MergeSharedMemoryAllocations()
    • MultiVersionBuffer()
    • PersistThreadblock()
    • PipelinePlanning()
    • RewriteWgmmaSync()
    • ThreadPartialSync()
    • ThreadSync()
    • VectorizeLoop()
    • WarpSpecialized()
    • WarpSpecializedPipeline()
    • get_pass_context()
• tilelang.utils package
  • Submodules
    • tilelang.utils.deprecated module
      • deprecated()
    • tilelang.utils.language module
      • array_reduce()
      • get_buffer_elems()
      • is_fragment()
      • is_global()
      • is_local()
      • is_shared()
      • is_shared_dynamic()
      • retrieve_func_from_module()
    • tilelang.utils.target module
      • check_cuda_availability()
      • check_hip_availability()
      • determine_target()
    • tilelang.utils.tensor module
      • TensorSupplyType
      • adapt_torch2tvm()
      • get_tensor_supply()
      • map_torch_type()
      • torch_assert_close()
  • Module contents

Submodules

• tilelang.env module
  • disable_cache()
  • enable_cache()
  • is_cache_enabled()

Module contents