[Triton-MLIR][BACKEND] Mark float to integer in Arithmetic Dialect as legal (#963)

2022-12-08 09:07:01 -08:00
parent c7cf9c6a32
commit 71c35bcf9c
2 changed files with 6 additions and 2 deletions
--- a/lib/Conversion/TritonToTritonGPU/TritonToTritonGPU.cpp
+++ b/lib/Conversion/TritonToTritonGPU/TritonToTritonGPU.cpp
@@ -114,6 +114,7 @@ void populateArithmeticPatternsAndLegality(
      GenericOpPattern<arith::TruncIOp>, GenericOpPattern<arith::TruncFOp>,
      GenericOpPattern<arith::ExtUIOp>, GenericOpPattern<arith::ExtSIOp>,
      GenericOpPattern<arith::ExtFOp>, GenericOpPattern<arith::SIToFPOp>,
+      GenericOpPattern<arith::FPToSIOp>, GenericOpPattern<arith::FPToUIOp>,
      GenericOpPattern<arith::UIToFPOp>>(typeConverter, context);
 }