diff --git a/third_party/intel/lib/TritonIntelGPUTransforms/Pipeliner/MatmulLoopPipeline.cpp b/third_party/intel/lib/TritonIntelGPUTransforms/Pipeliner/MatmulLoopPipeline.cpp index 031dd61824..2b6742fb42 100644 --- a/third_party/intel/lib/TritonIntelGPUTransforms/Pipeliner/MatmulLoopPipeline.cpp +++ b/third_party/intel/lib/TritonIntelGPUTransforms/Pipeliner/MatmulLoopPipeline.cpp @@ -84,11 +84,11 @@ static ttg::DotOperandEncodingAttr allTransitiveUsesHaveDotEncoding(Value val) { } /// Create a prefetch operation for the given load operation. -static void createPrefetchOp(scf::ForOp &forOp, tt::LoadOp loadOp, Value ptr) { +static void createPrefetchOp(scf::ForOp &forOp, tt::LoadOp loadOp) { OpBuilder builder(forOp); builder.setInsertionPoint(loadOp); auto prefetchOp = builder.create( - loadOp->getLoc(), ptr, loadOp.getCache(), loadOp.getEvict(), + loadOp->getLoc(), loadOp.getPtr(), loadOp.getCache(), loadOp.getEvict(), loadOp.getIsVolatile()); // inherit attributes from the load operation @@ -102,7 +102,7 @@ static void createPrefetchOps(scf::ForOp &forOp, assert(!loads.empty() && "Expecting at least one load operation"); for (const LoadDotOperand &loadOperand : loads) { tt::LoadOp loadOp = loadOperand.load; - createPrefetchOp(forOp, loadOp, loadOp.getPtr()); + createPrefetchOp(forOp, loadOp); } }