Lines Matching defs:writeOp

158   op->walk([&](Operation *writeOp) {
160     if (!isContiguousStore(writeOp))
162     Value vectorVal = nvgpu::getValueStored(writeOp);
165     Value storeBase = nvgpu::getMemrefOperand(writeOp);
201     copyToSharedMem.insert(writeOp);
208     Operation *writeOp = *copyToSharedMem.begin();
209     copyToSharedMem.remove(writeOp);
210     group.push_back(writeOp);
211     Operation *nextNode = writeOp;
241     for (Operation *writeOp : group) {
242       rewriter.setInsertionPoint(writeOp);
243       Value vectorVal = nvgpu::getValueStored(writeOp);
247       Value storeBase = nvgpu::getMemrefOperand(writeOp);
250           buildNumReadElements(rewriter, writeOp->getLoc(), readOp);
256           writeOp->getLoc(), nvgpu::DeviceAsyncTokenType::get(op->getContext()),
257           /*dst=*/storeBase, /*dstIndices=*/nvgpu::getIndices(writeOp),
274     for (Operation *writeOp : group)
275       rewriter.eraseOp(writeOp);