added relevant issue about dynamic shape config setting

nod-ai · Oct 10, 2023 · ec1ac74 · ec1ac74
1 parent 1876fa4
commit ec1ac74
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git 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
@@ -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: