Update

[ghstack-poisoned]
pytorch · Oct 23, 2024 · 9c78d0a · 9c78d0a
1 parent 7a264c1
commit 9c78d0a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_rb.py b/test/test_rb.py
@@ -429,7 +429,7 @@ def test_extend_sample_recompile(
         if datatype == "tensordict":
             pytest.skip("'tensordict' datatype is not currently supported.")
 
-        torch.compiler.reset()
+        torch._dynamo.reset_code_caches()
 
         storage_size = 10 * size
         rb = self._get_rb(