Home
last modified time | relevance | path

Searched refs:numGroups (Results 1 – 5 of 5) sorted by relevance

/llvm-project-15.0.7/mlir/test/Dialect/NVGPU/
H A Doptimize-shared-memory.mlir23 nvgpu.device_async_wait %1 { numGroups = 1 : i32}
43 nvgpu.device_async_wait %1 { numGroups = 1 : i32}
80 nvgpu.device_async_wait %1 { numGroups = 1 : i32}
136 nvgpu.device_async_wait %1 { numGroups = 1 : i32}
181 nvgpu.device_async_wait %1 { numGroups = 1 : i32}
207 nvgpu.device_async_wait %1 { numGroups = 1 : i32}
233 nvgpu.device_async_wait %1 { numGroups = 1 : i32}
H A Droundtrip.mlir30 // CHECK: nvgpu.device_async_wait %{{.*}} {numGroups = 1 : i32}
31 nvgpu.device_async_wait %token {numGroups = 1 : i32}
/llvm-project-15.0.7/mlir/lib/Conversion/NVGPUToNVVM/
H A DNVGPUToNVVM.cpp429 int32_t numGroups = adaptor.getNumGroups().value_or(0); in matchAndRewrite() local
430 rewriter.create<NVVM::CpAsyncWaitGroupOp>(op.getLoc(), numGroups); in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/NVGPU/IR/
H A DNVGPU.td226 OptionalAttr<I32Attr>:$numGroups);
/llvm-project-15.0.7/mlir/test/Conversion/NVGPUToNVVM/
H A Dnvgpu-to-nvvm.mlir268 nvgpu.device_async_wait %1 { numGroups = 1 : i32 }