From 1294157cbe44bc215bfdbe391f25f990ae649eae Mon Sep 17 00:00:00 2001
From: Anthony Cui <anthonycui@google.com>
Date: Thu, 11 Jul 2024 18:36:53 -0700
Subject: [PATCH] Create named subtests for each NCCL test

PiperOrigin-RevId: 651601366
---
 test/gpu/nccl_test.go | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/test/gpu/nccl_test.go b/test/gpu/nccl_test.go
index 67aa2ccff..8b9ee18ab 100644
--- a/test/gpu/nccl_test.go
+++ b/test/gpu/nccl_test.go
@@ -54,7 +54,8 @@ func TestNCCL(t *testing.T) {
 
 	ctx := context.Background()
 	for _, test := range testNames {
-		t.Logf("Running NCCL test: %s", test)
-		runNCCL(ctx, t, test)
+		t.Run(test, func(t *testing.T) {
+			runNCCL(ctx, t, test)
+		})
 	}
 }