tilelang.cpu.transform¶
CPU-specific transformation frontends.
Functions¶
Lower tl.atomic_*_elem_op intrinsics to serial read-modify-write. |
Package Contents¶
- tilelang.cpu.transform.LowerCPUAtomics()¶
Lower tl.atomic_*_elem_op intrinsics to serial read-modify-write.
CPU targets only: the pass rewrites scalar-path atomic intrinsics into plain BufferLoad/BufferStore RMW so that both CPU codegens (c and llvm) can consume them. Tile-region atomics are lowered separately by the cpu.AtomicAdd/cpu.AtomicReduce impls inside LowerTileOp.
- Returns:
fpass – The result pass
- Return type:
tvm.transform.Pass