Files
mscclpp/cmake/check_nvidia_gpu.cu
2025-07-28 09:59:20 +08:00

16 lines
274 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;
}