From ec1ac749c763284acf32f5e48e82f5a0c7bb0e2f Mon Sep 17 00:00:00 2001 From: brucekimrokcmu Date: Mon, 9 Oct 2023 18:53:27 -0700 Subject: [PATCH] added relevant issue about dynamic shape config setting --- examples/eager_mlp/mlp_eager_simple.py | 2 +- tests/dynamo/mninst_test.py | 5 ++++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/examples/eager_mlp/mlp_eager_simple.py b/examples/eager_mlp/mlp_eager_simple.py index d5bbb3443..947d0da6e 100644 --- a/examples/eager_mlp/mlp_eager_simple.py +++ b/examples/eager_mlp/mlp_eager_simple.py @@ -13,7 +13,7 @@ import torchvision.transforms as transforms import torchvision.datasets as datasets -torch._dynamo.config.dynamic_shapes = False +torch._dynamo.config.dynamic_shapes = False # TODO: https://github.com/nod-ai/SHARK-Turbine/issues/93 class MNISTDataLoader: diff --git a/tests/dynamo/mninst_test.py b/tests/dynamo/mninst_test.py index 6c599ed61..7374dde13 100644 --- a/tests/dynamo/mninst_test.py +++ b/tests/dynamo/mninst_test.py @@ -13,7 +13,10 @@ import torchvision.transforms as transforms import torchvision.datasets as datasets -torch._dynamo.config.dynamic_shapes = False + +import torch._dynamo.config + +torch._dynamo.config.dynamic_shapes = False # TODO: https://github.com/nod-ai/SHARK-Turbine/issues/93 class MNISTDataLoader: