Update dtype check in torch-to-tosa clamp op (llvm#2529)

As titled. --------- Co-authored-by: Ze Zhang <ze.zhang@getcruise.com>
knwng · Oct 24, 2023 · 7cb2db6 · 7cb2db6
1 parent 6d74e8c
commit 7cb2db6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/Conversion/TorchToTosa/TorchToTosa.cpp b/lib/Conversion/TorchToTosa/TorchToTosa.cpp
@@ -3986,7 +3986,7 @@ LogicalResult ConvertAtenOp<AtenClampOp>::matchAndRewrite(
 
   IntegerAttr min_int, max_int;
   FloatAttr min_fp, max_fp;
-  if (selfType.getElementType().isa<mlir::FloatType>()) {
+  if (op.getMin().getType().isa<Torch::FloatType>()) {
     double fp_min, fp_max;
     if (!matchPattern(op.getMin(), m_TorchConstantFloat(&fp_min)))
       return rewriter.notifyMatchFailure(