Skip to content

Commit 0ae4695

Browse files
committed
Make ModelAlsoUsingSubmoduleParamsDirectly and InterModuleParamReuse xpass
1 parent 0101465 commit 0ae4695

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/unit/autogram/test_engine.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -124,6 +124,8 @@
124124
32,
125125
marks=mark.filterwarnings("ignore:There is a performance drop"),
126126
),
127+
(ModuleFactory(ModelAlsoUsingSubmoduleParamsDirectly), 32),
128+
(ModuleFactory(InterModuleParamReuse), 32),
127129
(ModuleFactory(FreeParam), 32),
128130
(ModuleFactory(NoFreeParam), 32),
129131
param(ModuleFactory(Cifar10Model), 16, marks=mark.slow),
@@ -193,8 +195,6 @@ def test_compute_gramian_with_weird_modules(
193195
"factory",
194196
[
195197
ModuleFactory(ModelUsingSubmoduleParamsDirectly),
196-
ModuleFactory(ModelAlsoUsingSubmoduleParamsDirectly),
197-
ModuleFactory(InterModuleParamReuse),
198198
],
199199
)
200200
@mark.parametrize("batch_size", [1, 3, 32])

0 commit comments

Comments
 (0)