[Gemini] test step-tensor mapping using repeated_computed_layers.py (#2127)

2025-09-10 21:40:02 +00:00 · 2022-12-13 16:34:10 +08:00
parent 8fac837679
commit deee317b0f
3 changed files with 14 additions and 2 deletions
--- a/tests/components_to_test/init.py
+++ b/tests/components_to_test/init.py
@@ -4,10 +4,15 @@ from . import (
    hanging_param_model,
    inline_op_model,
    nested_model,
-    repeated_computed_layer,
+    repeated_computed_layers,
    resnet,
    simple_net,
 )
 from .utils import run_fwd_bwd

 from . import albert    # isort:skip
+
+__all__ = [
+    'bert', 'gpt2', 'hanging_param_model', 'inline_op_model', 'nested_model', 'repeated_computed_layers', 'resnet',
+    'simple_net', 'run_fwd_bwd', 'albert'
+]
--- a/tests/components_to_test/repeated_computed_layers.py
+++ b/tests/components_to_test/repeated_computed_layers.py