fix

cruise-automation · Jul 16, 2024 · bf3aa99 · bf3aa99
1 parent 6ba811e
commit bf3aa99
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/lib/Conversion/TorchToTcp/TcpCustomOp.cpp b/lib/Conversion/TorchToTcp/TcpCustomOp.cpp
@@ -154,14 +154,15 @@ class ConvertAtenFakeQuantizePerTensorAffineTensorQparamsOp
     helper.addIntAttr("quant_max", op.getQuantMax());
 
     // scale
-    auto scaleTy = adaptor.getScale().dyn_cast<RankedTensorType>();
+    auto scaleTy = adaptor.getScale().getType().dyn_cast<RankedTensorType>();
     if (!scaleTy || scaleTy.getShape().size() != 1 ||
         scaleTy.getNumElements() != 1)
       return rewriter.notifyMatchFailure(op, "Unsupported scale type or size");
     helper.addOperand("scale", adaptor.getScale());
 
     // zero_point
-    auto zeroPointTy = adaptor.getZeroPoint().dyn_cast<RankedTensorType>();
+    auto zeroPointTy =
+        adaptor.getZeroPoint().getType().dyn_cast<RankedTensorType>();
     if (!zeroPointTy || zeroPointTy.getShape().size() != 1 ||
         zeroPointTy.getNumElements() != scaleTy.getNumElements())
       return rewriter.notifyMatchFailure(op,
@@ -188,13 +189,14 @@ class ConvertAtenFakeQuantizePerChannelAffineOp
     helper.addIntAttr("quant_max", op.getQuantMax());
 
     // scale
-    auto scaleTy = adaptor.getScale().dyn_cast<RankedTensorType>();
+    auto scaleTy = adaptor.getScale().getType().dyn_cast<RankedTensorType>();
     if (!scaleTy || scaleTy.getShape().size() != 1)
       return rewriter.notifyMatchFailure(op, "Unsupported scale type or size");
     helper.addOperand("scale", adaptor.getScale());
 
     // zero_point
-    auto zeroPointTy = adaptor.getZeroPoint().dyn_cast<RankedTensorType>();
+    auto zeroPointTy =
+        adaptor.getZeroPoint().getType().dyn_cast<RankedTensorType>();
     if (!zeroPointTy || zeroPointTy.getShape().size() != 1 ||
         zeroPointTy.getNumElements() != scaleTy.getNumElements())
       return rewriter.notifyMatchFailure(op,