1 //===- TestVectorReductionToSPIRVDotProd.cpp - Test reduction to dot prod -===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #include "mlir/Conversion/VectorToSPIRV/VectorToSPIRV.h" 10 #include "mlir/Dialect/Arith/IR/Arith.h" 11 #include "mlir/Dialect/Func/IR/FuncOps.h" 12 #include "mlir/Dialect/SPIRV/IR/SPIRVDialect.h" 13 #include "mlir/Dialect/Vector/IR/VectorOps.h" 14 #include "mlir/Pass/Pass.h" 15 #include "mlir/Pass/PassManager.h" 16 #include "mlir/Transforms/GreedyPatternRewriteDriver.h" 17 18 namespace mlir { 19 namespace { 20 21 struct TestVectorReductionToSPIRVDotProd 22 : PassWrapper<TestVectorReductionToSPIRVDotProd, 23 OperationPass<func::FuncOp>> { 24 MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID( 25 TestVectorReductionToSPIRVDotProd) 26 27 StringRef getArgument() const final { 28 return "test-vector-reduction-to-spirv-dot-prod"; 29 } 30 31 StringRef getDescription() const final { 32 return "Test lowering patterns that converts vector.reduction to SPIR-V " 33 "integer dot product ops"; 34 } 35 36 void getDependentDialects(DialectRegistry ®istry) const override { 37 registry.insert<arith::ArithDialect, func::FuncDialect, spirv::SPIRVDialect, 38 vector::VectorDialect>(); 39 } 40 41 void runOnOperation() override { 42 RewritePatternSet patterns(&getContext()); 43 populateVectorReductionToSPIRVDotProductPatterns(patterns); 44 (void)applyPatternsGreedily(getOperation(), std::move(patterns)); 45 } 46 }; 47 48 } // namespace 49 50 namespace test { 51 void registerTestVectorReductionToSPIRVDotProd() { 52 PassRegistration<TestVectorReductionToSPIRVDotProd>(); 53 } 54 } // namespace test 55 } // namespace mlir 56