Attempt to fix stable ABI calls

Author: Sam Anklesaria

src/libtorchaudio/rnnt/compute.cpp

@@ -1,12 +1,14 @@
 #include <torch/csrc/stable/library.h>
+#include <torch/csrc/inductor/aoti_torch/c/shim.h>
+
 
 STABLE_TORCH_LIBRARY_FRAGMENT(torchaudio, m) {
   m.def(
-      "torchaudio::rnnt_loss(Tensor logits,"
+      "rnnt_loss(Tensor logits,"
       "Tensor targets,"
       "Tensor logit_lengths,"
       "Tensor target_lengths,"
       "int blank,"
       "float clamp,"
-      "bool fused_log_softmax) -> (Tensor, Tensor?)");
+      "bool fused_log_softmax) -> (Tensor, Tensor)");
 }

src/libtorchaudio/rnnt/cpu/compute.cpp

@@ -206,7 +206,7 @@ void boxed_compute(StableIValue* stack, uint64_t num_args, uint64_t num_outputs)
 }
 
 STABLE_TORCH_LIBRARY_IMPL(torchaudio, CPU, m) {
-  m.impl("torchaudio::rnnt_loss", &boxed_compute);
+  m.impl("rnnt_loss", &boxed_compute);
 }
 
 } // namespace cpu

src/torchaudio/functional/functional.py

@@ -1763,7 +1763,7 @@ def _fix_waveform_shape(
 class RnntLoss(torch.autograd.Function):
     @staticmethod
     def forward(ctx, *args):
-        output, saved = torch.ops.torchaudio.rnnt_loss(*args)
+        output, saved = torch.ops.torchaudio.rnnt_loss.default(*args)
         ctx.save_for_backward(saved)
         return output