fixes

microsoft · Nov 11, 2023 · b5323da · b5323da
1 parent 74adad4
commit b5323da
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git a/python/test/_cpp/proxy_test.cpp b/python/test/_cpp/proxy_test.cpp
@@ -1,6 +1,7 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT license.
 
+#include <cuda.h>
 #include <nanobind/nanobind.h>
 #include <nanobind/stl/shared_ptr.h>
 #include <nanobind/stl/vector.h>

diff --git a/test/mscclpp-test/allgather_test.cu b/test/mscclpp-test/allgather_test.cu
@@ -1,9 +1,10 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT license.
 
+#include <cuda_runtime.h>
+
 #include <algorithm>
 #include <mscclpp/concurrency.hpp>
-#include <mscclpp/gpu.hpp>
 #include <string>
 
 #include "common.hpp"

diff --git a/test/mscclpp-test/common.cc b/test/mscclpp-test/common.cc
@@ -3,6 +3,7 @@
 
 #include "common.hpp"
 
+#include <cuda.h>
 #include <getopt.h>
 #include <libgen.h>
 #include <mpi.h>
@@ -15,7 +16,6 @@
 #include <fstream>
 #include <iomanip>
 #include <iostream>
-#include <mscclpp/gpu.hpp>
 #include <mscclpp/utils.hpp>
 #include <nlohmann/json.hpp>
 #include <sstream>

diff --git a/test/mscclpp-test/sendrecv_test.cu b/test/mscclpp-test/sendrecv_test.cu
@@ -6,7 +6,7 @@
 #include <cstring>
 #include <iostream>
 #include <mscclpp/concurrency.hpp>
-#include <mscclpp/gpu_utils.hpp>
+#include <mscclpp/cuda_utils.hpp>
 #include <mscclpp/semaphore.hpp>
 #include <mscclpp/sm_channel.hpp>
 #include <string>