From 6903420310fe28da99ba8bcdabad4521f4a0e285 Mon Sep 17 00:00:00 2001 From: Jason Lowe-Power Date: Wed, 27 Apr 2022 16:48:29 -0700 Subject: [PATCH] tests: Add more hierarchies to traffic gen tests This adds all of the hierarchies supported in the standard library. We can soon move to using a different build target and run all hierarchies! Change-Id: Ic065a679ea34c3bb2f71b3b133806d240039fbb5 Signed-off-by: Jason Lowe-Power --- .../traffic_gen/configs/simple_traffic_run.py | 27 ++++++++++++++++++- 1 file changed, 26 insertions(+), 1 deletion(-) diff --git a/tests/gem5/traffic_gen/configs/simple_traffic_run.py b/tests/gem5/traffic_gen/configs/simple_traffic_run.py index 7c264cefe9..45de2e7601 100644 --- a/tests/gem5/traffic_gen/configs/simple_traffic_run.py +++ b/tests/gem5/traffic_gen/configs/simple_traffic_run.py @@ -133,6 +133,24 @@ def cache_factory(cache_class: str): l2_assoc="4", num_l2_banks=1, ) + elif cache_class == "MIExample": + from gem5.components.cachehierarchies.ruby.mi_example_cache_hierarchy import ( + MIExampleCacheHierarchy, + ) + + return MIExampleCacheHierarchy( + size="16KiB", + assoc=8, + ) + elif cache_class == "CHIL1": + from gem5.components.cachehierarchies.chi.private_l1_cache_hierarchy import ( + PrivateL1CacheHierarchy, + ) + + return PrivateL1CacheHierarchy( + size="16KiB", + assoc=8, + ) else: raise ValueError(f"The cache class {cache_class} is not supported.") @@ -163,7 +181,14 @@ parser.add_argument( "cache_class", type=str, help="The cache class to import and instantiate.", - choices=["NoCache", "PrivateL1", "PrivateL1PrivateL2", "MESITwoLevel"], + choices=[ + "NoCache", + "PrivateL1", + "PrivateL1PrivateL2", + "MESITwoLevel", + "MIExample", + "CHIL1", + ], ) parser.add_argument(