Skip to content

Commit a2b97d8

Browse files
committed
Change compile backend to eager
1 parent 524bcae commit a2b97d8

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tests/encoders/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -213,7 +213,7 @@ def test_compile(self):
213213
encoder = encoder.eval().to(default_device)
214214

215215
torch.compiler.reset()
216-
compiled_encoder = torch.compile(encoder, fullgraph=True, dynamic=True)
216+
compiled_encoder = torch.compile(encoder, fullgraph=True, dynamic=True, backend="eager")
217217

218218
if encoder._is_torch_compilable:
219219
compiled_encoder(sample)

tests/models/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -232,7 +232,7 @@ def test_compile(self):
232232
model = model.eval().to(default_device)
233233

234234
torch.compiler.reset()
235-
compiled_model = torch.compile(model, fullgraph=True, dynamic=True)
235+
compiled_model = torch.compile(model, fullgraph=True, dynamic=True, backend="eager")
236236

237237
with torch.inference_mode():
238238
compiled_model(sample)

0 commit comments

Comments
 (0)