Files
mscclpp/cmake/check_nvidia_gpu.cu
Changho Hwang 544ff0c21d ROCm support (#213)
Co-authored-by: Binyang Li <binyli@microsoft.com>
2023-11-24 16:41:56 +08:00

16 lines
288 B
Plaintext

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT license.
#include <cuda_runtime.h>
__global__ void kernel() {}
int main() {
int cnt;
cudaError_t err = cudaGetDeviceCount(&cnt);
if (err != cudaSuccess || cnt == 0) {
return 1;
}
return 0;
}