--- ./c10/core/DispatchKeySet.cpp_orig	2025-01-13 03:29:29.294615812 -0600
+++ ./c10/core/DispatchKeySet.cpp	2025-01-13 02:34:03.954614236 -0600
@@ -59,7 +59,7 @@
 constexpr DispatchKeySet nested_dispatch_keyset =
     DispatchKeySet(
         {DispatchKey::AutogradNestedTensor, DispatchKey::NestedTensor}) |
-    DispatchKeySet(DispatchKeySet::RAW, full_backend_mask);
+    DispatchKeySet(DispatchKeySet::RAWDATA, full_backend_mask);
 
 DispatchKeySet getRuntimeDispatchKeySet(DispatchKey t) {
   TORCH_INTERNAL_ASSERT(t != DispatchKey::Undefined);
@@ -70,7 +70,7 @@
       // getRuntimeDispatchKeySet() expects to return a keyset of runtime
       // dispatch keys, like AutogradCPU, but that requires having backend bits.
       return autograd_dispatch_keyset |
-          DispatchKeySet(DispatchKeySet::RAW, full_backend_mask);
+          DispatchKeySet(DispatchKeySet::RAWDATA, full_backend_mask);
     case DispatchKey::CompositeImplicitAutograd:
       return math_dispatch_keyset;
     case DispatchKey::CompositeImplicitAutogradNestedTensor:
@@ -139,7 +139,7 @@
       return DispatchKeySet(DispatchKey::PrivateUse3);
     case DispatchKey::AutogradNestedTensor:
       return DispatchKeySet(DispatchKey::NestedTensor) |
-          DispatchKeySet(DispatchKeySet::RAW, full_backend_mask);
+          DispatchKeySet(DispatchKeySet::RAWDATA, full_backend_mask);
     case DispatchKey::AutogradOther:
       return autogradother_backends;
     default:
