Skip to content

Commit 6bedf56

Browse files
committed
Make ModelAlsoUsingSubmoduleParamsDirectly and InterModuleParamReuse xpass
1 parent a8f5380 commit 6bedf56

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
@@ -126,6 +126,8 @@
126126
32,
127127
marks=mark.filterwarnings("ignore:There is a performance drop"),
128128
),
129+
(ModuleFactory(ModelAlsoUsingSubmoduleParamsDirectly), 32),
130+
(ModuleFactory(InterModuleParamReuse), 32),
129131
(ModuleFactory(FreeParam), 32),
130132
(ModuleFactory(NoFreeParam), 32),
131133
param(ModuleFactory(Cifar10Model), 16, marks=mark.slow),
@@ -222,8 +224,6 @@ def test_compute_gramian_with_weird_modules(
222224
"factory",
223225
[
224226
ModuleFactory(ModelUsingSubmoduleParamsDirectly),
225-
ModuleFactory(ModelAlsoUsingSubmoduleParamsDirectly),
226-
ModuleFactory(InterModuleParamReuse),
227227
],
228228
)
229229
@mark.parametrize("batch_size", [1, 3, 32])

0 commit comments

Comments
 (0)