Searched refs:matrixB (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/NVGPU/IR/ |
| H A D | NVGPU.td | 90 PredOpTrait<"matrixA and matrixB have same element type",
113 let arguments = (ins AnyVector:$matrixA, AnyVector:$matrixB,
119 `(` $matrixA`,` $matrixB`,` $matrixC `)` attr-dict
120 `:` `(` type($matrixA) `,` type($matrixB) `,` type($matrixC) `)` `->` type($res)
|
| /llvm-project-15.0.7/flang/unittests/Optimizer/Builder/Runtime/ |
| H A D | TransformationalTest.cpp | 67 mlir::Value matrixB = firBuilder->create<fir::UndefOp>(loc, seqTy); in TEST_F() local
68 fir::runtime::genMatmul(*firBuilder, loc, matrixA, matrixB, result); in TEST_F()
|
| /llvm-project-15.0.7/mlir/test/Dialect/NVGPU/ |
| H A D | invalid.mlir | 115 // expected-error @+1 {{op failed to verify that matrixA and matrixB have same element type}}
|
| /llvm-project-15.0.7/flang/lib/Lower/ |
| H A D | IntrinsicCall.cpp | 3294 mlir::Value matrixB = fir::getBase(matrixTmpB); in genMatmul() local
3305 fir::runtime::genMatmul(builder, loc, resultIrBox, matrixA, matrixB); in genMatmul()
|