tests: move to c++

stack-info: PR: #568, branch: aws-nslick/stack/15
aws · Sep 4, 2024 · e7c5e0c · e7c5e0c
1 parent f071592
commit e7c5e0c
Show file tree

Hide file tree

Showing 16 changed files with 34 additions and 31 deletions.
diff --git a/tests/functional/Makefile.am b/tests/functional/Makefile.am
@@ -11,18 +11,18 @@ LDADD = $(top_builddir)/src/libinternal_net_plugin.la $(MPI_LIBS) $(CUDA_LIBS)
 CC = $(MPICC)
 CXX = $(MPICXX)
 
-noinst_HEADERS = test-common.h
+noinst_HEADERS = test-common.hpp
 
 if ENABLE_TESTS
 bin_PROGRAMS = nccl_connection nccl_message_transfer ring
 noinst_PROGRAMS = cuda_check
 endif
 
-nccl_connection_SOURCES = nccl_connection.c
-nccl_message_transfer_SOURCES = nccl_message_transfer.c
-ring_SOURCES = ring.c
+nccl_connection_SOURCES = nccl_connection.cc
+nccl_message_transfer_SOURCES = nccl_message_transfer.cc
+ring_SOURCES = ring.cc
 
-cuda_check_SOURCES = cuda_check.c
+cuda_check_SOURCES = cuda_check.cc
 # Override the LDADD for this check to avoid the -lcudart used by the
 # other tests, since the purpose of this test is to make sure we
 # didn't leak direct cuda dependencies into the plugin.

diff --git a/tests/functional/cuda_check.c → tests/functional/cuda_check.cc b/tests/functional/cuda_check.c → tests/functional/cuda_check.cc
diff --git a/tests/functional/nccl_connection.c → tests/functional/nccl_connection.cc b/tests/functional/nccl_connection.c → tests/functional/nccl_connection.cc
@@ -8,7 +8,7 @@
 
 #include "config.h"
 
-#include "test-common.h"
+#include "test-common.hpp"
 
 int main(int argc, char* argv[])
 {

diff --git a/tests/functional/nccl_message_transfer.c → tests/functional/nccl_message_transfer.cc b/tests/functional/nccl_message_transfer.c → tests/functional/nccl_message_transfer.cc
@@ -9,9 +9,10 @@
 
 #include "config.h"
 
-#include "test-common.h"
+#include "test-common.hpp"
 
-#define PROC_NAME_IDX(i) (i * MPI_MAX_PROCESSOR_NAME)
+template <typename T>
+constexpr T PROC_NAME_IDX(T i) { return i * MPI_MAX_PROCESSOR_NAME; };
 
 int main(int argc, char* argv[])
 {
@@ -57,6 +58,7 @@ int main(int argc, char* argv[])
 	size_t recv_sizes[] = {4 * 1024, 16 * 1024, 1024 * 1024,
 	                       4 * 1024, 16 * 1024, 1024 * 1024,
 	                       5 * 1024, 17 * 1024, 2 * 1024 * 1024};
+	test_nccl_properties_t props = {};
 
 	/* For grouped recvs */
 	int tag = 1;
@@ -133,7 +135,6 @@ int main(int argc, char* argv[])
 		goto exit;
 	}
 
-	test_nccl_properties_t props = {};
 
 	/* Get Properties for the device */
 	for (int dev = 0; dev < ndev; dev++) {
@@ -398,7 +399,7 @@ int main(int argc, char* argv[])
 	MPI_Finalize();
 	NCCL_OFI_INFO(NCCL_NET, "Test completed successfully for rank %d", rank);
 
-exit:;
+exit:
 
 	ncclResult_t close_res = ncclSuccess;
 

diff --git a/tests/functional/ring.c → tests/functional/ring.cc b/tests/functional/ring.c → tests/functional/ring.cc
@@ -4,7 +4,7 @@
 
 #include "config.h"
 
-#include "test-common.h"
+#include "test-common.hpp"
 
 #define PROC_NAME_IDX(i) (i * MPI_MAX_PROCESSOR_NAME)
 

diff --git a/tests/functional/test-common.h → tests/functional/test-common.hpp b/tests/functional/test-common.h → tests/functional/test-common.hpp
diff --git a/tests/unit/Makefile.am b/tests/unit/Makefile.am
@@ -7,7 +7,7 @@
 AM_CPPFLAGS = -I$(top_srcdir)/include
 LDADD = $(top_builddir)/src/libinternal_net_plugin.la
 
-noinst_HEADERS = test-common.h
+noinst_HEADERS = test-common.hpp
 
 noinst_PROGRAMS = \
 	deque \
@@ -21,12 +21,13 @@ noinst_PROGRAMS = \
 
 TESTS = $(noinst_PROGRAMS)
 
-show_tuner_decisions_SOURCES = show_tuner_decisions.c
+show_tuner_decisions_SOURCES = show_tuner_decisions.cc
 show_tuner_decisions_LDADD = $(top_builddir)/src/libinternal_tuner_plugin.la
 
-deque_SOURCES = deque.c
-freelist_SOURCES = freelist.c
-msgbuff_SOURCES = msgbuff.c
-scheduler_SOURCES = scheduler.c
-ep_addr_list_SOURCES = ep_addr_list.c
-mr_SOURCES = mr.c
+idpool_SOURCES = idpool.cc
+deque_SOURCES = deque.cc
+freelist_SOURCES = freelist.cc
+msgbuff_SOURCES = msgbuff.cc
+scheduler_SOURCES = scheduler.cc
+ep_addr_list_SOURCES = ep_addr_list.cc
+mr_SOURCES = mr.cc
diff --git a/tests/unit/deque.c → tests/unit/deque.cc b/tests/unit/deque.c → tests/unit/deque.cc
@@ -6,19 +6,18 @@
 
 #include <stdio.h>
 
-#include "test-common.h"
+#include "test-common.hpp"
 #include "nccl_ofi_deque.h"
 
 int main(int argc, char *argv[])
 {
-	const size_t num_elem = 11;
+	constexpr size_t num_elem = 11;
 	struct elem_t {
 		nccl_ofi_deque_elem_t de;
 		int v;
 	} elems[num_elem];
 
 	nccl_ofi_deque_elem_t *deque_elem;
-	int ret;
 	size_t i;
 	for (i = 0; i < num_elem; ++i) {
 		elems[i].v = i;
@@ -27,7 +26,7 @@ int main(int argc, char *argv[])
 	ofi_log_function = logger;
 
 	nccl_ofi_deque_t *deque;
-	ret = nccl_ofi_deque_init(&deque);
+	int ret = nccl_ofi_deque_init(&deque);
 	if (ret) {
 		NCCL_OFI_WARN("deque_init failed: %d", ret);
 		exit(1);
@@ -51,7 +50,7 @@ int main(int argc, char *argv[])
 	for (i = 0; i < num_elem; ++i) {
 		int expected = (i == 0 ? elems[num_elem-1].v : elems[i-1].v);
 		ret = nccl_ofi_deque_remove_front(deque, &deque_elem);
-		if (ret || deque_elem == NULL) {
+		if (ret || deque_elem == nullptr) {
 			NCCL_OFI_WARN("remove_front unexpectedly failed: %d", ret);
 			exit(1);
 		}
@@ -62,7 +61,7 @@ int main(int argc, char *argv[])
 		}
 	}
 	ret = nccl_ofi_deque_remove_front(deque, &deque_elem);
-	if (ret != 0 || deque_elem != NULL) {
+	if (ret != 0 || deque_elem != nullptr) {
 		NCCL_OFI_WARN("remove_front from empty deque unexpectedly succeeded");
 		exit(1);
 	}

diff --git a/tests/unit/ep_addr_list.c → tests/unit/ep_addr_list.cc b/tests/unit/ep_addr_list.c → tests/unit/ep_addr_list.cc
@@ -6,7 +6,7 @@
 
 #include <stdio.h>
 
-#include "test-common.h"
+#include "test-common.hpp"
 #include "nccl_ofi_ep_addr_list.h"
 
 static void insert_addresses(nccl_ofi_ep_addr_list_t *ep_addr_list, size_t num_addr, int ep_num)

diff --git a/tests/unit/freelist.c → tests/unit/freelist.cc b/tests/unit/freelist.c → tests/unit/freelist.cc
@@ -6,7 +6,7 @@
 
 #include <stdio.h>
 
-#include "test-common.h"
+#include "test-common.hpp"
 #include "nccl_ofi_freelist.h"
 
 void *simple_base;

diff --git a/tests/unit/idpool.c → tests/unit/idpool.cc b/tests/unit/idpool.c → tests/unit/idpool.cc
@@ -6,7 +6,7 @@
 
 #include <stdio.h>
 
-#include "test-common.h"
+#include "test-common.hpp"
 #include "nccl_ofi_idpool.h"
 #include "nccl_ofi_math.h"
 

diff --git a/tests/unit/mr.c → tests/unit/mr.cc b/tests/unit/mr.c → tests/unit/mr.cc
@@ -6,7 +6,7 @@
 
 #include <stdlib.h>
 
-#include "test-common.h"
+#include "test-common.hpp"
 #include "nccl_ofi_mr.h"
 
 bool test_lookup_impl(nccl_ofi_mr_cache_t *cache, void *addr, size_t size,

diff --git a/tests/unit/msgbuff.c → tests/unit/msgbuff.cc b/tests/unit/msgbuff.c → tests/unit/msgbuff.cc
@@ -8,7 +8,7 @@
 
 #include "nccl_ofi_msgbuff.h"
 
-#include "test-common.h"
+#include "test-common.hpp"
 
 int main(int argc, char *argv[])
 {

diff --git a/tests/unit/scheduler.c → tests/unit/scheduler.cc b/tests/unit/scheduler.c → tests/unit/scheduler.cc
@@ -11,7 +11,7 @@
 
 #include "nccl-headers/net.h"
 
-#include "test-common.h"
+#include "test-common.hpp"
 #include "nccl_ofi_scheduler.h"
 
 int create_multiplexed(size_t size,

diff --git a/tests/unit/show_tuner_decisions.c → tests/unit/show_tuner_decisions.cc b/tests/unit/show_tuner_decisions.c → tests/unit/show_tuner_decisions.cc
diff --git a/tests/unit/test-common.h → tests/unit/test-common.hpp b/tests/unit/test-common.h → tests/unit/test-common.hpp
@@ -11,6 +11,7 @@
 #include "nccl_ofi.h"
 #include "nccl_ofi_log.h"
 
+namespace {
 static inline void logger(ncclDebugLogLevel level, unsigned long flags, const char *filefunc,
 			  int line, const char *fmt, ...)
 {
@@ -39,5 +40,6 @@ static inline void logger(ncclDebugLogLevel level, unsigned long flags, const ch
 	printf("\n");
 	va_end(vargs);
 }
+}
 
 #endif // End TEST_COMMON_H_