tilelang.language package

Subpackages

• tilelang.language.tir package
  • Submodules
    • tilelang.language.tir.entry module
      • macro()
      • prim_func()
    • tilelang.language.tir.ir module
      • grid()
      • parallel()
      • serial()
      • thread_binding()
      • unroll()
      • vectorized()
    • tilelang.language.tir.op module
      • TVMBackendAllocWorkspace()
      • TVMBackendFreeWorkspace()
      • abs()
      • acos()
      • acosh()
      • address_of()
      • all()
      • any()
      • anylist_getitem()
      • anylist_resetitem()
      • anylist_setitem_call_cpacked()
      • anylist_setitem_call_packed()
      • asin()
      • asinh()
      • assume()
      • atan()
      • atan2()
      • atanh()
      • bitwise_and()
      • bitwise_not()
      • bitwise_or()
      • bitwise_xor()
      • call_cpacked()
      • call_cpacked_lowered()
      • call_extern()
      • call_intrin()
      • call_llvm_intrin()
      • call_llvm_pure_intrin()
      • call_packed()
      • call_packed_lowered()
      • call_pure_extern()
      • call_tir()
      • ceil()
      • ceildiv()
      • clz()
      • comm_reducer()
      • copysign()
      • cos()
      • cosh()
      • create_barriers()
      • div()
      • end_profile_intrinsic()
      • erf()
      • exp()
      • exp10()
      • exp2()
      • floor()
      • floordiv()
      • floormod()
      • fmod()
      • hypot()
      • if_then_else()
      • indexdiv()
      • indexmod()
      • infinity()
      • isfinite()
      • isinf()
      • isnan()
      • isnullptr()
      • ldexp()
      • likely()
      • log()
      • log10()
      • log1p()
      • log2()
      • lookup_param()
      • max_value()
      • min_value()
      • mma_fill()
      • mma_store()
      • nearbyint()
      • nextafter()
      • popcount()
      • pow()
      • power()
      • ptx_arrive_barrier()
      • ptx_arrive_barrier_expect_tx()
      • ptx_commit_group()
      • ptx_cp_async()
      • ptx_cp_async_barrier()
      • ptx_cp_async_bulk()
      • ptx_init_barrier_thread_count()
      • ptx_ldmatrix()
      • ptx_mma()
      • ptx_mma_sp()
      • ptx_wait_barrier()
      • ptx_wait_group()
      • q_multiply_shift()
      • q_multiply_shift_per_axis()
      • reinterpret()
      • ret()
      • round()
      • rsqrt()
      • shift_left()
      • shift_right()
      • sigmoid()
      • sin()
      • sinh()
      • sqrt()
      • start_profile_intrinsic()
      • tan()
      • tanh()
      • trace()
      • trunc()
      • truncdiv()
      • truncmod()
      • tvm_access_ptr()
      • tvm_bmma_sync()
      • tvm_check_return()
      • tvm_fill_fragment()
      • tvm_load_matrix_sync()
      • tvm_mfma()
      • tvm_mfma_store()
      • tvm_mma_sync()
      • tvm_rdna_wmma()
      • tvm_rdna_wmma_store()
      • tvm_stack_alloca()
      • tvm_stack_make_array()
      • tvm_stack_make_shape()
      • tvm_storage_sync()
      • tvm_store_matrix_sync()
      • tvm_struct_get()
      • tvm_struct_set()
      • tvm_thread_allreduce()
      • tvm_thread_invariant()
      • tvm_throw_last_error()
      • tvm_tuple()
      • tvm_warp_activemask()
      • tvm_warp_shuffle()
      • tvm_warp_shuffle_down()
      • tvm_warp_shuffle_up()
      • type_annotation()
      • undef()
      • vectorcombine()
      • vectorhigh()
      • vectorlow()
      • vscale()
  • Module contents

Submodules

• tilelang.language.allocate module
  • alloc_fragment()
  • alloc_local()
  • alloc_shared()
  • alloc_var()
• tilelang.language.builtin module
  • CreateListofMBarrierOp()
  • CreateTMADescriptorOp()
  • FenceProxyAsyncOp()
  • GetMBarrierOp()
  • MBarrierExpectTX()
  • MBarrierWaitParity()
  • NoSetMaxNReg()
  • SetMaxNReg()
  • TMALoadOp()
  • TMAStoreArrive()
  • TMAStoreWait()
  • WaitWgmma()
• tilelang.language.copy module
  • buffer_load_to_tile_region()
  • buffer_region_to_tile_region()
  • buffer_to_tile_region()
  • c2d_im2col()
  • copy()
  • region()
• tilelang.language.customize module
  • atomic_add()
  • atomic_addx2()
  • clamp()
  • dp4a()
  • reshape()
  • view()
• tilelang.language.fill module
  • clear()
  • fill()
• tilelang.language.frame module
  • FrameStack
    • FrameStack.get_value()
    • FrameStack.has_value()
    • FrameStack.pop()
    • FrameStack.push()
    • FrameStack.top()
  • LetFrame
    • LetFrame.Current()
    • LetFrame.get_value()
    • LetFrame.has_value()
  • get_let_value()
  • has_let_value()
• tilelang.language.gemm module
  • gemm()
• tilelang.language.kernel module
  • FrameStack
    • FrameStack.pop()
    • FrameStack.push()
    • FrameStack.top()
  • Kernel()
  • KernelLaunchFrame
    • KernelLaunchFrame.Current()
    • KernelLaunchFrame.blocks
    • KernelLaunchFrame.get_block_binding()
    • KernelLaunchFrame.get_block_bindings()
    • KernelLaunchFrame.get_block_extent()
    • KernelLaunchFrame.get_num_threads()
    • KernelLaunchFrame.get_thread_binding()
    • KernelLaunchFrame.get_thread_bindings()
    • KernelLaunchFrame.get_thread_extent()
    • KernelLaunchFrame.num_threads
    • KernelLaunchFrame.threads
  • get_block_binding()
  • get_block_bindings()
  • get_thread_binding()
  • get_thread_bindings()
• tilelang.language.memscope module
• tilelang.language.parallel module
  • Parallel()
• tilelang.language.pipeline module
  • Pipelined()
• tilelang.language.print module
  • print()
• tilelang.language.proxy module
  • BaseTensorProxy
    • BaseTensorProxy.default_align
    • BaseTensorProxy.default_offset_factor
    • BaseTensorProxy.default_scope
    • BaseTensorProxy.from_ptr()
  • BufferProxy
    • BufferProxy.from_ptr()
  • FragmentBufferProxy
    • FragmentBufferProxy.default_scope
  • LocalBufferProxy
    • LocalBufferProxy.default_scope
  • SharedBufferProxy
    • SharedBufferProxy.default_scope
  • TensorProxy
  • make_tensor()
  • ptr()
• tilelang.language.reduce module
  • reduce()
  • reduce_absmax()
  • reduce_abssum()
  • reduce_max()
  • reduce_min()
  • reduce_sum()

Module contents

The language interface for tl programs.

tilelang.language.annotate_layout(layout_map)

tilelang.language.import_source(source: Optional[str] = None)

tilelang.language.symbolic(name: str, dtype: str = 'int32')

tilelang.language.use_swizzle(panel_size: int, order: str = 'row', enable: boolean = True)