Fix some tests (especially compile with fullgraph=True on Python<3.11) (#38319)
* fix tests * better fix for python<3.11 * fixes * style
This commit is contained in:
@@ -4461,6 +4461,7 @@ class ModelTesterMixin:
|
||||
del loss
|
||||
|
||||
model = torch.compile(model, fullgraph=True, mode="reduce-overhead")
|
||||
|
||||
# forward compilation
|
||||
set_seed(42)
|
||||
loss = model(**inputs).loss
|
||||
|
||||
Reference in New Issue
Block a user