Files
mscclpp/cmake/check_amd_gpu.hip
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 <hip/hip_runtime.h>
__global__ void kernel() {}
int main() {
int cnt;
hipError_t err = hipGetDeviceCount(&cnt);
if (err != hipSuccess || cnt == 0) {
return 1;
}
return 0;
}