From 00311f45e1b6de9726ec34ef9e0a2d2e18119249 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Mon, 29 Jan 2024 17:41:43 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 src/acc/libsmm_acc/kernels/smm_acc_predict.py |  2 +-
 src/acc/libsmm_acc/predict/predict_collect.py | 12 ++++++------
 src/acc/libsmm_acc/predict/predict_train.py   |  6 +++---
 src/acc/opencl/smm/tune_multiply.py           | 16 ++++++++++------
 4 files changed, 20 insertions(+), 16 deletions(-)

diff --git a/src/acc/libsmm_acc/kernels/smm_acc_predict.py b/src/acc/libsmm_acc/kernels/smm_acc_predict.py
index 111b9a3c7df..b65f48cdaa5 100644
--- a/src/acc/libsmm_acc/kernels/smm_acc_predict.py
+++ b/src/acc/libsmm_acc/kernels/smm_acc_predict.py
@@ -42,7 +42,7 @@
         "size_b",
         "size_c",
         # 'nblks', 'nthreads',  # constant value for largeDB, since the grouping is always = 16
-        "sm_desired"
+        "sm_desired",
         # 'warps_per_blk', 'nwarps',  # linearly dependent on threads, nthreads
         # 'ru_param_stack_unroll_factor',  # constant values for each algo
     ],
diff --git a/src/acc/libsmm_acc/predict/predict_collect.py b/src/acc/libsmm_acc/predict/predict_collect.py
index 9dee95323af..ab41ebe1de2 100755
--- a/src/acc/libsmm_acc/predict/predict_collect.py
+++ b/src/acc/libsmm_acc/predict/predict_collect.py
@@ -105,12 +105,12 @@ def read_log_file(log_folder, m, n, k):
                         "threads": match.group(9),
                         "grouping": match.group(10),
                         "minblocks": match.group(11),
-                        "tile_m": match.group(5)
-                        if match.group(5) is not None
-                        else None,
-                        "tile_n": match.group(6)
-                        if match.group(6) is not None
-                        else None,
+                        "tile_m": (
+                            match.group(5) if match.group(5) is not None else None
+                        ),
+                        "tile_n": (
+                            match.group(6) if match.group(6) is not None else None
+                        ),
                         "w": match.group(7) if match.group(7) is not None else None,
                         "v": match.group(8) if match.group(8) is not None else None,
                         "perf (Gflop/s)": match.group(12),
diff --git a/src/acc/libsmm_acc/predict/predict_train.py b/src/acc/libsmm_acc/predict/predict_train.py
index 7370cc8990c..cf2b3845202 100755
--- a/src/acc/libsmm_acc/predict/predict_train.py
+++ b/src/acc/libsmm_acc/predict/predict_train.py
@@ -1184,9 +1184,9 @@ def get_predive_model_performances(
 
         perf_chosen_idx = [np.argmax(y_pred[idx_mnk])]
         perf_effective = y_true.iloc[idx_mnk].iloc[perf_chosen_idx].values.item()
-        predictive_model_perf_scaled[
-            (m, n, k)
-        ] = perf_effective  # 'scaled' between 0 and 1
+        predictive_model_perf_scaled[(m, n, k)] = (
+            perf_effective  # 'scaled' between 0 and 1
+        )
 
     predictive_model_perf = dict(
         zip(
diff --git a/src/acc/opencl/smm/tune_multiply.py b/src/acc/opencl/smm/tune_multiply.py
index cc39b1009db..a6fd8e64f02 100755
--- a/src/acc/opencl/smm/tune_multiply.py
+++ b/src/acc/opencl/smm/tune_multiply.py
@@ -367,9 +367,11 @@ def merge_jsons(self, filenames):
                 key = (device, data["TYPEID"], data["M"], data["N"], data["K"])
                 value = (
                     data["S"] if "S" in data else 0,  # pseudo key component
-                    data["GFLOPS"]
-                    if "GFLOPS" in data and not self.args.nogflops
-                    else 0,
+                    (
+                        data["GFLOPS"]
+                        if "GFLOPS" in data and not self.args.nogflops
+                        else 0
+                    ),
                     data["BS"],
                     data["BM"],
                     data["BN"],
@@ -525,9 +527,11 @@ def save_final_config(self, configuration, final=True):
                 gflops = data["GFLOPS"] if data and "GFLOPS" in data else 0
                 filename = os.path.join(
                     self.args.jsondir,
-                    "{}-{}gflops.json".format(self.args.label, round(gflops))
-                    if 0 < gflops
-                    else "{}.json".format(self.args.label),
+                    (
+                        "{}-{}gflops.json".format(self.args.label, round(gflops))
+                        if 0 < gflops
+                        else "{}.json".format(self.args.label)
+                    ),
                 )
                 try:
                     os.rename(filedot, filename)