Skip to content

Commit 846421b

Browse files
committed
Add ConvertMathToXeVMPass to the pipeline.
1 parent 6fe5a12 commit 846421b

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

mlir/lib/Dialect/GPU/Pipelines/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ add_mlir_dialect_library(MLIRGPUPipelines
1616
MLIRGPUToNVVMTransforms
1717
MLIRIndexToLLVM
1818
MLIRMathToLLVM
19+
MLIRMathToXeVM
1920
MLIRNVGPUToNVVM
2021
MLIRNVVMToLLVM
2122
MLIRReconcileUnrealizedCasts

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

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
#include "mlir/Conversion/VectorToSCF/VectorToSCF.h"
1919
#include "mlir/Conversion/XeGPUToXeVM/XeGPUToXeVM.h"
2020
#include "mlir/Conversion/XeVMToLLVM/XeVMToLLVM.h"
21+
#include "mlir/Conversion/MathToXeVM/MathToXeVM.h"
2122
#include "mlir/Dialect/Func/IR/FuncOps.h"
2223
#include "mlir/Dialect/GPU/IR/GPUDialect.h"
2324
#include "mlir/Dialect/GPU/Pipelines/Passes.h"
@@ -76,6 +77,7 @@ void buildGpuPassPipeline(OpPassManager &pm,
7677
pm.addNestedPass<gpu::GPUModuleOp>(createCSEPass());
7778
pm.addNestedPass<gpu::GPUModuleOp>(xegpu::createXeGPUVectorLinearize());
7879
}
80+
pm.addNestedPass<gpu::GPUModuleOp>(createConvertMathToXeVM());
7981
pm.addNestedPass<gpu::GPUModuleOp>(createConvertXeGPUToXeVMPass());
8082
{
8183
ConvertGpuOpsToLLVMSPVOpsOptions gpuToLLVMSPVOptions;

0 commit comments

Comments
 (0)