From ef507494291813512dd9f67da3e3ae3622788ebf Mon Sep 17 00:00:00 2001 From: Adam Siemieniuk Date: Thu, 3 Oct 2024 11:29:31 +0200 Subject: [PATCH] Refactor after rebase --- lib/TPP/GPU/GpuPipeline.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/lib/TPP/GPU/GpuPipeline.cpp b/lib/TPP/GPU/GpuPipeline.cpp index 7cbba1986..06f238aaf 100644 --- a/lib/TPP/GPU/GpuPipeline.cpp +++ b/lib/TPP/GPU/GpuPipeline.cpp @@ -65,9 +65,9 @@ llvm::cl::list llvm::cl::CommaSeparated); // Control GPU vectorization. -llvm::cl::opt gpuVectorize("gpu-vectorize", - llvm::cl::desc("Vectorize GPU kernel"), - llvm::cl::init(false)); +llvm::cl::opt gpuVector("gpu-vector", + llvm::cl::desc("Vectorize GPU kernel"), + llvm::cl::init(false)); namespace mlir { namespace tpp { @@ -187,7 +187,7 @@ struct GpuPipeline : public tpp::impl::GpuPipelineBase, pm.addPass(createTileConsumerAndFuseProducers(threadTileOptions)); pm.addPass(createCleanup()); - if (gpuVectorize) { + if (gpuVector) { // Early reduction dimension splitting is incompatible with // Linalg to XeGPU lowering that expects full GEMM. // For now, enable only with other vectorization passes.