Skip to content

Commit 35c5dc8

Browse files
committed
addressed review comments
1 parent 39b3942 commit 35c5dc8

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

recipes_source/torch_export_aoti_python.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -79,11 +79,6 @@
7979
aot_compile_options.update({"max_autotune": True})
8080
else:
8181
device = "cpu"
82-
# We need to turn off the below optimizations to support batch_size = 16,
83-
# which is treated like a special case
84-
# https://github.com/pytorch/pytorch/pull/116152
85-
torch.backends.mkldnn.set_flags(False)
86-
torch.backends.nnpack.set_flags(False)
8782

8883
model = model.to(device=device)
8984
example_inputs = (torch.randn(2, 3, 224, 224, device=device),)

0 commit comments

Comments
 (0)