Searched defs:num_stages (Results 1 – 3 of 3) sorted by relevance
| /llvm-project/mlir/test/Examples/NVGPU/ |
| H A D | Ch5.py | 82 num_stages, argument
119 def initialize(a_tma: TMA, b_tma: TMA, num_stages): argument
139 def switch_phase(stage, phase, num_stages): argument
155 num_stages, argument
176 num_stages, argument
254 def gemm_warp_specialized(a, b, d, num_stages): argument
|
| H A D | Ch4.py | 77 num_stages, argument
117 def initialize(a_tma: TMA, b_tma: TMA, num_stages): argument
135 def prologue(mbar_group: Mbarriers, a_tma: TMA, b_tma: TMA, num_stages): argument
149 def mainloop(mbar_group: Mbarriers, a_tma: TMA, b_tma: TMA, num_stages): argument
260 def gemm_multistage(a, b, d, num_stages): argument
|
| /llvm-project/mlir/test/Integration/GPU/CUDA/sm90/python/tools/ |
| H A D | matmulBuilder.py | 140 num_stages=3, argument
172 num_stages=3, argument
672 generate_matmul_multistage( input_type=np.float16, output_type=np.float32, M=4096, N=4096, K=4096, BLOCK_M=128, BLOCK_N=128, BLOCK_K=64, num_stages=3, ) global() argument
[all...] |