@@ -33,7 +33,7 @@ class SlowNodeDetector:
33
33
"""
34
34
35
35
def __init__ (
36
- self , path , sensors , num_nodes , pct , spn , rpn , plot_rank_breakdowns ):
36
+ self , path , sensors , num_nodes , pct , spn , rpn , plot_rank_breakdowns , use_clstr ):
37
37
# Create empty dicts for storing data
38
38
self .__rank_times = {}
39
39
self .__rank_breakdowns = {}
@@ -52,7 +52,7 @@ def __init__(
52
52
self .__temperature_analysis_available = True if self .__sensors_output_file is not None else False
53
53
self .__plot_rank_breakdowns = plot_rank_breakdowns
54
54
self .__num_ranks = 0
55
- self .__use_clustering = use_clustering
55
+ self .__use_clustering = use_clstr
56
56
57
57
# Initialize outliers
58
58
self .__slow_ranks = {}
@@ -663,10 +663,11 @@ def main():
663
663
path = filepath ,
664
664
sensors = sensors_filepath ,
665
665
num_nodes = args .num_nodes ,
666
- threshold_percentage = args .threshold ,
667
- sockets_per_node = args .spn ,
668
- ranks_per_node = args .rpn ,
669
- plot_rank_breakdowns = args .plot_all_ranks )
666
+ pct = args .threshold ,
667
+ spn = args .spn ,
668
+ rpn = args .rpn ,
669
+ plot_rank_breakdowns = args .plot_all_ranks ,
670
+ use_clstr = args .use_clustering )
670
671
671
672
slowNodeDetector .detect ()
672
673
slowNodeDetector .createHostfile ()
0 commit comments