Add check on any_extend output type

Author: adprasad-nvidia

llvm/lib/Target/AArch64/AArch64ISelLowering.cpp

@@ -22219,10 +22219,13 @@ static SDValue performExtendCombine(SDNode *N,
   // ...)), which is what this pattern would otherwise be lowered to.
   if (N->getOpcode() == ISD::ANY_EXTEND &&
       N->getOperand(0).getOpcode() == ISD::BSWAP &&
-      N->getOperand(0).getValueType() == MVT::i16) {
+      N->getOperand(0).getValueType() == MVT::i16 &&
+      (N->getValueType(0) == MVT::i32 ||
+       N->getValueType(0) == MVT::i64)) {
     SDNode *BswapNode = N->getOperand(0).getNode();
+    SDLoc DL(N);
     SDValue NewAnyExtend =
-        DAG.getNode(ISD::ANY_EXTEND, SDLoc(BswapNode), N->getValueType(0),
+        DAG.getNode(ISD::ANY_EXTEND, DL, N->getValueType(0),
                     BswapNode->getOperand(0));
     return DAG.getNode(AArch64ISD::REV16, SDLoc(N), N->getValueType(0),
                        NewAnyExtend);