Index: tools/perf/core/benchmark_finders.py |
diff --git a/tools/perf/core/benchmark_finders.py b/tools/perf/core/benchmark_finders.py |
index 6705b16cb537e547f02baec107b2329be9ad9917..4cb79e6651e8f2f633316d7af342359d4dee133d 100644 |
--- a/tools/perf/core/benchmark_finders.py |
+++ b/tools/perf/core/benchmark_finders.py |
@@ -44,8 +44,16 @@ def GetBenchmarkNamesForFile(top_level_dir, benchmark_file_dir): |
def GetAllPerfBenchmarks(): |
- benchmarks_dir = path_util.GetPerfBenchmarksDir() |
- top_level_dir = os.path.join(benchmarks_dir, '..') |
return discover.DiscoverClasses( |
- benchmarks_dir, top_level_dir, benchmark_module.Benchmark, |
+ start_dir=path_util.GetPerfBenchmarksDir(), |
+ top_level_dir=path_util.GetPerfDir(), |
+ base_class=benchmark_module.Benchmark, |
+ index_by_class_name=True).values() |
+ |
+ |
+def GetAllContribBenchmarks(): |
+ return discover.DiscoverClasses( |
+ start_dir=path_util.GetPerfContribDir(), |
+ top_level_dir=path_util.GetPerfDir(), |
+ base_class=benchmark_module.Benchmark, |
index_by_class_name=True).values() |