Lines Matching defs:contract

143 gpuMmaUnrollOrder(vector::ContractionOp contract) {
146   for (auto [index, iter] : llvm::enumerate(contract.getIteratorTypes())) {
153   for (AffineExpr expr : contract.getIndexingMapsArray()[0].getResults()) {
157   for (auto [index, iter] : llvm::enumerate(contract.getIteratorTypes())) {
163   for (auto [index, iter] : llvm::enumerate(contract.getIteratorTypes())) {
175   if (auto contract = dyn_cast<vector::ContractionOp>(op)) {
176     int64_t contractRank = contract.getIteratorTypes().size();
209       // only to operations that need unrolling (connected to the contract).
215       // types, in which case we will see an `arith.ext_` between the contract
243     auto contract = dyn_cast<vector::ContractionOp>(op);
244     if (!contract)
246     return gpuMmaUnrollOrder(contract);