tilelang.carver.arch

Submodules

• tilelang.carver.arch.arch_base
• tilelang.carver.arch.cdna
• tilelang.carver.arch.cpu
• tilelang.carver.arch.cuda
• tilelang.carver.arch.driver

Functions

get_arch([target])
auto_infer_current_arch()

Package Contents

tilelang.carver.arch.get_arch(target='cuda')

Parameters:

target (Union[str, tvm.target.Target])

Return type:

arch_base.TileDevice

tilelang.carver.arch.auto_infer_current_arch()

Return type:

arch_base.TileDevice