Lines Matching refs:state
143 AnalysisState &state) { in bufferizeInPlace() argument
145 for (OpResult result : state.getAliasingOpResult(operand)) in bufferizeInPlace()
331 const AnalysisState &state) { in isInplaceMemoryWrite() argument
333 if (!state.bufferizesToMemoryWrite(opOperand)) in isInplaceMemoryWrite()
371 static bool isMemoryWrite(Value value, const AnalysisState &state) { in isMemoryWrite() argument
375 auto bufferizableOp = state.getOptions().dynCastBufferizableOp(value); in isMemoryWrite()
378 return bufferizableOp.isMemoryWrite(opResult, state); in isMemoryWrite()
424 AnalysisState &state, const BufferizationAliasInfo &aliasInfo) { in hasReadAfterWriteInterference() argument
425 const BufferizationOptions &options = state.getOptions(); in hasReadAfterWriteInterference()
430 if (isMemoryWrite(uWrite->get(), state)) in hasReadAfterWriteInterference()
450 SetVector<Value> lastWrites = state.findLastPrecedingWrite(uRead->get()); in hasReadAfterWriteInterference()
522 if (bufferizableOp.isNotConflicting(uRead, uConflictingWrite, state)) in hasReadAfterWriteInterference()
528 if (bufferizableOp.isNotConflicting(uRead, uConflictingWrite, state)) in hasReadAfterWriteInterference()
563 state.getAliasingOpResult(*uConflictingWrite); in hasReadAfterWriteInterference()
583 const AnalysisState &state) { in getAliasingInplaceWrites() argument
587 if (isInplaceMemoryWrite(use, aliasInfo, state)) in getAliasingInplaceWrites()
595 const AnalysisState &state) { in getAliasingReads() argument
599 if (state.bufferizesToMemoryRead(use)) in getAliasingReads()
634 OpOperand &operand, const DominanceInfo &domInfo, AnalysisState &state, in wouldCreateReadAfterWriteInterference() argument
639 getAliasingReads(usesRead, operand.get(), aliasInfo, state); in wouldCreateReadAfterWriteInterference()
640 getAliasingInplaceWrites(usesWrite, operand.get(), aliasInfo, state); in wouldCreateReadAfterWriteInterference()
641 for (OpResult result : state.getAliasingOpResult(operand)) { in wouldCreateReadAfterWriteInterference()
642 getAliasingReads(usesRead, result, aliasInfo, state); in wouldCreateReadAfterWriteInterference()
643 getAliasingInplaceWrites(usesWrite, result, aliasInfo, state); in wouldCreateReadAfterWriteInterference()
645 if (!checkConsistencyOnly && state.bufferizesToMemoryWrite(operand)) in wouldCreateReadAfterWriteInterference()
648 return hasReadAfterWriteInterference(usesRead, usesWrite, domInfo, state, in wouldCreateReadAfterWriteInterference()
660 const OneShotAnalysisState &state) { in hasPrecedingAliasingNonWritableTensor() argument
665 if (!state.isWritable(nextVal)) in hasPrecedingAliasingNonWritableTensor()
675 state.getAliasingOpOperand(opResult); in hasPrecedingAliasingNonWritableTensor()
687 OneShotAnalysisState &state, bool checkConsistencyOnly = false) { in wouldCreateWriteToNonWritableBuffer() argument
690 getAliasingInplaceWrites(usesWrite, operand.get(), aliasInfo, state); in wouldCreateWriteToNonWritableBuffer()
691 for (OpResult result : state.getAliasingOpResult(operand)) { in wouldCreateWriteToNonWritableBuffer()
692 getAliasingInplaceWrites(usesWrite, result, aliasInfo, state); in wouldCreateWriteToNonWritableBuffer()
694 if (!checkConsistencyOnly && state.bufferizesToMemoryWrite(operand)) in wouldCreateWriteToNonWritableBuffer()
702 aliasInfo, state)) in wouldCreateWriteToNonWritableBuffer()
715 OneShotAnalysisState &state, const DominanceInfo &domInfo) { in bufferizableInPlaceAnalysisImpl() argument
717 wouldCreateWriteToNonWritableBuffer(operand, aliasInfo, state) || in bufferizableInPlaceAnalysisImpl()
718 wouldCreateReadAfterWriteInterference(operand, domInfo, state, aliasInfo); in bufferizableInPlaceAnalysisImpl()
723 aliasInfo.bufferizeInPlace(operand, state); in bufferizableInPlaceAnalysisImpl()
748 OneShotAnalysisState &state, in inPlaceAnalysis() argument
763 if (auto bufferizableOp = state.getOptions().dynCastBufferizableOp(op)) in inPlaceAnalysis()
765 state, domInfo))) in inPlaceAnalysis()
781 OneShotAnalysisState &state, in inPlaceAnalysis() argument
793 return inPlaceAnalysis(ops, aliasInfo, state, domInfo, analysisFuzzerSeed); in inPlaceAnalysis()
799 AnalysisState &state) { in equivalenceAnalysis() argument
801 if (auto bufferizableOp = state.getOptions().dynCastBufferizableOp(op)) in equivalenceAnalysis()
805 bufferizableOp.getAliasingOpOperand(opResult, state)) in equivalenceAnalysis()
806 if (state.isInPlace(*opOperand)) in equivalenceAnalysis()
807 if (bufferizableOp.bufferRelation(opResult, state) == in equivalenceAnalysis()
816 AnalysisState &state) { in equivalenceAnalysis() argument
826 equivalenceAnalysis(ops, aliasInfo, state); in equivalenceAnalysis()
832 AnalysisState &state, in checkAliasInfoConsistency() argument
834 const BufferizationOptions &options = state.getOptions(); in checkAliasInfoConsistency()
841 opOperand, domInfo, state, aliasInfo, in checkAliasInfoConsistency()
862 AnalysisState &state) { in annotateOpsWithBufferizationMarkers() argument
864 if (auto bufferizableOp = state.getOptions().dynCastBufferizableOp(op)) in annotateOpsWithBufferizationMarkers()
895 assertDestinationPassingStyle(Operation *op, AnalysisState &state, in assertDestinationPassingStyle() argument
902 !state.getOptions().isOpAllowed(returnOp)) in assertDestinationPassingStyle()
942 OneShotAnalysisState &state) { in analyzeOp() argument
944 BufferizationAliasInfo &aliasInfo = state.getAliasInfo(); in analyzeOp()
946 static_cast<const OneShotBufferizationOptions &>(state.getOptions()); in analyzeOp()
949 assert((state.hasDialectState(func::FuncDialect::getDialectNamespace()) || in analyzeOp()
953 if (failed(checkAliasInfoConsistency(op, domInfo, state, aliasInfo))) in analyzeOp()
957 if (failed(inPlaceAnalysis(op, aliasInfo, state, domInfo, in analyzeOp()
960 equivalenceAnalysis(op, aliasInfo, state); in analyzeOp()
966 failed(assertDestinationPassingStyle(op, state, aliasInfo, newOps)); in analyzeOp()
970 state.gatherYieldedTensors(op); in analyzeOp()
971 state.gatherUndefinedTensorUses(op); in analyzeOp()
979 failedAnalysis |= failed(bufferizableOp.verifyAnalysis(state)); in analyzeOp()
984 annotateOpsWithBufferizationMarkers(op, aliasInfo, state); in analyzeOp()
992 OneShotAnalysisState state(op, options); in runOneShotBufferize() local