From 948eae48c39b06eeb994e6807fe919df278f004d Mon Sep 17 00:00:00 2001 From: Cloud Han Date: Mon, 4 Sep 2023 01:31:09 +0800 Subject: [PATCH] Address formatting --- examples/WORKSPACE.bazel | 5 ++++- examples/nccl/nccl.bzl | 2 +- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/examples/WORKSPACE.bazel b/examples/WORKSPACE.bazel index 51346df9..7b6b5cb3 100644 --- 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 index 8a8553c8..e2758f27 100644 --- 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: