diff --git a/mlir/lib/Conversion/VectorToGPU/VectorToGPU.cpp b/mlir/lib/Conversion/VectorToGPU/VectorToGPU.cpp --- a/mlir/lib/Conversion/VectorToGPU/VectorToGPU.cpp +++ b/mlir/lib/Conversion/VectorToGPU/VectorToGPU.cpp @@ -191,12 +191,26 @@ return gpu::MMAElementwiseOp::ADDF; if (isa(op)) return gpu::MMAElementwiseOp::MULF; + if (isa(op)) + return gpu::MMAElementwiseOp::SUBF; if (isa(op)) return gpu::MMAElementwiseOp::MAXF; if (isa(op)) return gpu::MMAElementwiseOp::MINF; if (isa(op)) return gpu::MMAElementwiseOp::DIVF; + if (isa(op)) + return gpu::MMAElementwiseOp::ADDI; + if (isa(op)) + return gpu::MMAElementwiseOp::MULI; + if (isa(op)) + return gpu::MMAElementwiseOp::SUBI; + if (isa(op)) + return gpu::MMAElementwiseOp::DIVS; + if (isa(op)) + return gpu::MMAElementwiseOp::DIVU; + if (isa(op)) + return gpu::MMAElementwiseOp::NEGATEF; return llvm::None; }