Skip to content

Commit 3072c1c

Browse files
committed
Fix a small logic.
1 parent 75d5547 commit 3072c1c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

mlir/lib/Dialect/GPU/Pipelines/GPUToXeVMPipeline.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,8 @@ void buildGpuPassPipeline(OpPassManager &pm,
7070
pm.addNestedPass<gpu::GPUModuleOp>(createCanonicalizerPass());
7171
pm.addNestedPass<gpu::GPUModuleOp>(createCSEPass());
7272
}
73-
if (options.xegpuOpLevel == "subgroup") {
73+
if (options.xegpuOpLevel == "subgroup" ||
74+
options.xegpuOpLevel == "workgroup") {
7475
pm.addNestedPass<gpu::GPUModuleOp>(xegpu::createXeGPUPropagateLayout());
7576
pm.addNestedPass<gpu::GPUModuleOp>(xegpu::createXeGPUSubgroupDistribute());
7677
pm.addNestedPass<gpu::GPUModuleOp>(createCanonicalizerPass());

0 commit comments

Comments
 (0)