From 9c78d0a6a8ee9baba8406e77acaa688c247f8c6d Mon Sep 17 00:00:00 2001
From: Kurt Mohler <kurtamohler@gmail.com>
Date: Wed, 23 Oct 2024 14:11:39 -0700
Subject: [PATCH] Update

[ghstack-poisoned]
---
 test/test_rb.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/test_rb.py b/test/test_rb.py
index 9db96e5d8c0..d451bcef9c4 100644
--- 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(