fix none type check

llvm · Nov 1, 2024 · 7dcdd6d · 7dcdd6d
1 parent 712d87f
commit 7dcdd6d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/Conversion/TorchToTMTensor/TorchToTMTensor.cpp b/lib/Conversion/TorchToTMTensor/TorchToTMTensor.cpp
@@ -1726,7 +1726,7 @@ class ConvertAtenScaledDotProductAttentionOp
     }
 
     // Broadcast the batch dimensions of the mask:
-    if (mask) {
+    if (!isa<Torch::NoneType>(mask.getType())) {
       auto maskTy = cast<RankedTensorType>(mask.getType());
       int64_t rank = maskTy.getRank();
       bool needsBroadcast = false;