From 11597979c4161886890c2125e886089adf272c33 Mon Sep 17 00:00:00 2001 From: Thomas Viehmann Date: Wed, 18 Dec 2024 10:18:12 +0100 Subject: [PATCH] remat and recomp are exclusive --- thunder/tests/test_nvfuser_remat.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/thunder/tests/test_nvfuser_remat.py b/thunder/tests/test_nvfuser_remat.py index 690c4f93bc..46cdffddc5 100644 --- a/thunder/tests/test_nvfuser_remat.py +++ b/thunder/tests/test_nvfuser_remat.py @@ -308,7 +308,7 @@ def func(t0): return t3, t4 t0 = make_tensor(2, 2, dtype=torch.float32, device=device) - compiled_func = executor.make_callable(func) + compiled_func = executor.make_callable(func, enable_saved_for_backward_recomputation=False) _ = compiled_func(t0) traces = thunder.last_traces(compiled_func) trace = traces[0] @@ -369,7 +369,10 @@ def func( try: compiled_func = thunder.jit( - func, executors=(torch_compile_cat_ex, thunder.nvfuser_executor), use_rematerialization=True + func, + executors=(torch_compile_cat_ex, thunder.nvfuser_executor), + use_rematerialization=True, + enable_saved_for_backward_recomputation=False, ) _ = compiled_func( t0,