llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix check in test/Conversion/TorchToArith/basic.mlir 

Co-authored-by: zjgarvey <47986913+zjgarvey@users.noreply.github.com>
pull/3750/head
Giacomo Serafini 2024-11-20 08:58:01 +00:00 committed by GitHub
parent c6fdab1dd8
commit 74effb24e3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/Conversion/TorchToArith/basic.mlir
View File

@ -242,7 +242,7 @@ func.func @torch.aten.mul.int(%arg0: !torch.int, %arg1: !torch.int) -> !torch.in
// CHECK-DAG: %[[LHS_F64:.*]] = torch_c.to_f64 %[[LHS]]
// CHECK-DAG: %[[RHS_I64:.*]] = torch_c.to_i64 %[[RHS]]
// CHECK: %[[RHS_F64:.*]] = arith.sitofp %[[RHS_I64]] : i64 to f64
// CHECK: %[[MUL:.*]] = arith.mulf %[[LHS_F64:.*]], [[RHS_F64:.*]] : f64
// CHECK: %[[MUL:.*]] = arith.mulf %[[LHS_F64]], [[RHS_F64]] : f64
// CHECK: %[[OUT:.*]] = torch_c.from_f64 %[[MUL:.*]]
// CHECK: return %[[OUT:.*]] : !torch.float
func.func @torch.aten.mul.float_int(%arg0: !torch.float, %arg1: !torch.int) -> !torch.float {