tilelang.utils package

Submodules

• tilelang.utils.deprecated module
  • deprecated()
• tilelang.utils.language module
  • array_reduce()
  • 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
    • TensorSupplyType.Auto
    • TensorSupplyType.Integer
    • TensorSupplyType.Normal
    • TensorSupplyType.One
    • TensorSupplyType.Randn
    • TensorSupplyType.Uniform
    • TensorSupplyType.Zero
  • adapt_torch2tvm()
  • get_tensor_supply()
  • map_torch_type()
  • torch_assert_close()

Module contents

The profiler and convert to torch utils