fix bug

llvm · Nov 2, 2023 · 6724064 · 6724064
1 parent 5ed54af
commit 6724064
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/Dialect/Torch/IR/TorchOps.cpp b/lib/Dialect/Torch/IR/TorchOps.cpp
@@ -174,8 +174,8 @@ static Value getScalarFloatValue(Value input, Location loc,
     return nullptr;
 
   Type inputDtype = inputTensorType.getOptionalDtype();
-  if (!inputDtype || !inputDtype.isF16() || !inputDtype.isF32() ||
-      !inputDtype.isF64())
+  if (!inputDtype ||
+      (!inputDtype.isF16() && !inputDtype.isF32() && !inputDtype.isF64()))
     return nullptr;
 
   std::optional<unsigned> inputRank = getTensorRank(input);