Address formatting

bazel-contrib · Sep 3, 2023 · 948eae4 · 948eae4
1 parent 0216f8c
commit 948eae4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/examples/WORKSPACE.bazel b/examples/WORKSPACE.bazel
@@ -36,8 +36,11 @@ http_archive(
     name = "nccl-tests",
     add_prefix = "nccl-tests",
     build_file = "@rules_cuda_examples//nccl:nccl-tests.BUILD",
+    patch_args = [
+        "--directory=nccl-tests",
+        "-p1",
+    ],
     patches = ["@rules_cuda_examples//nccl:nccl-tests-clang.patch"],
-    patch_args = ["--directory=nccl-tests", "-p1"],
     sha256 = "946adb84f63aec66aea7aab9739d41df81c24f783e85fba6328ba243cfc057e0",
     strip_prefix = "nccl-tests-1a5f551ffd6e3271982b03a9d5653a3f6ba545fa",
     urls = ["https://github.com/nvidia/nccl-tests/archive/1a5f551ffd6e3271982b03a9d5653a3f6ba545fa.tar.gz"],

diff --git a/examples/nccl/nccl.bzl b/examples/nccl/nccl.bzl
@@ -13,7 +13,7 @@ def if_cuda_clang(if_true, if_false = []):
         "//conditions:default": if_false,
     })
 
-def nccl_primitive(name, hdrs = [],  deps = [], use_bf16 = True):
+def nccl_primitive(name, hdrs = [], deps = [], use_bf16 = True):
     ops = ["sum", "prod", "min", "max", "premulsum", "sumpostdiv"]
     datatypes = ["i8", "u8", "i32", "u32", "i64", "u64", "f16", "f32", "f64"]
     if use_bf16: