Formatter corrections

Signed-off-by: Gregory Shimansky <[email protected]>

Author: gshimansky

python/setup.py

@@ -592,7 +592,8 @@ def get_platform_dependent_src_path(subdir):
       )(*version.split('.'))))
 download_and_copy(
     name="cupti", src_path=get_platform_dependent_src_path("include"), dst_path="include",
-    variable="TRITON_CUPTI_INCLUDE_PATH", version=NVIDIA_TOOLCHAIN_VERSION["cupti"], url_func=lambda system, arch, version:
+    variable="TRITON_CUPTI_INCLUDE_PATH", version=NVIDIA_TOOLCHAIN_VERSION["cupti"],
+    url_func=lambda system, arch, version:
     ((lambda version_major, version_minor1, version_minor2:
       f"https://anaconda.org/nvidia/cuda-cupti-dev/{version}/download/{system}-{arch}/cuda-cupti-dev-{version}-0.tar.bz2"
       if int(version_major) >= 12 and int(version_minor1) >= 5 else

python/src/main.cc

@@ -6,9 +6,12 @@ namespace py = pybind11;
 
 #define EXPAND(x) x
 #define FOR_EACH_1(MACRO, X) MACRO(X)
-#define FOR_EACH_2(MACRO, X, ...) MACRO(X) EXPAND(FOR_EACH_1(MACRO, __VA_ARGS__))
-#define FOR_EACH_3(MACRO, X, ...) MACRO(X) EXPAND(FOR_EACH_2(MACRO, __VA_ARGS__))
-#define FOR_EACH_4(MACRO, X, ...) MACRO(X) EXPAND(FOR_EACH_3(MACRO, __VA_ARGS__))
+#define FOR_EACH_2(MACRO, X, ...)                                              \
+  MACRO(X) EXPAND(FOR_EACH_1(MACRO, __VA_ARGS__))
+#define FOR_EACH_3(MACRO, X, ...)                                              \
+  MACRO(X) EXPAND(FOR_EACH_2(MACRO, __VA_ARGS__))
+#define FOR_EACH_4(MACRO, X, ...)                                              \
+  MACRO(X) EXPAND(FOR_EACH_3(MACRO, __VA_ARGS__))
 
 #define FOR_EACH_NARG(...) FOR_EACH_NARG_(__VA_ARGS__, FOR_EACH_RSEQ_N())
 #define FOR_EACH_NARG_(...) EXPAND(FOR_EACH_ARG_N(__VA_ARGS__))
@@ -19,7 +22,8 @@ namespace py = pybind11;
 #define CONCATENATE1(x, y) x##y
 
 #define FOR_EACH(MACRO, ...)                                                   \
-  EXPAND(CONCATENATE(FOR_EACH_, FOR_EACH_NARG_HELPER(__VA_ARGS__))(MACRO, __VA_ARGS__))
+  EXPAND(CONCATENATE(FOR_EACH_,                                                \
+                     FOR_EACH_NARG_HELPER(__VA_ARGS__))(MACRO, __VA_ARGS__))
 #define FOR_EACH_NARG_HELPER(...) FOR_EACH_NARG(__VA_ARGS__)
 
 // New macro to remove parentheses

python/triton/runtime/build.py

@@ -39,7 +39,7 @@ def _cc_cmd(cc, src, out, include_dirs, library_dirs, libraries):
     else:
         cc_cmd = [cc, src, "-O3", "-shared"]
         if os.name != "nt":
-          cc_cmd += ["fPIC"]
+            cc_cmd += ["fPIC"]
         cc_cmd += [f'-l{lib}' for lib in libraries]
         cc_cmd += [f"-L{dir}" for dir in library_dirs]
         cc_cmd += [f"-I{dir}" for dir in include_dirs]