Give up inplace sanity check for a specific test

Lightning-AI · Sep 25, 2024 · d7ff759 · d7ff759
1 parent 9f005ca
commit d7ff759
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/thunder/tests/test_jit_general.py b/thunder/tests/test_jit_general.py
@@ -760,7 +760,9 @@ def sample(logits):
 
  with device:
  model.set_kv_cache(batch_size=1)
- tom = thunder.jit(model, executors=executors) # , disable_torch_autograd_support=True
+ tom = thunder.jit(
+ model, executors=executors, disable_inplace_copy_check=True
+ ) # , disable_torch_autograd_support=True
 
  # kv cache prefill
  thunder_logits_1 = tom(x, torch.tensor([0, 1], device=device))