How to detect whether GPU is AMD or NVIDIA from inside HIP code

Question

I'm currently writing a HIP equivalent to NVIDIA's deviceQuery sample code. I want my code to work on both AMD and NVIDIA hardware. Now, hipDeviceProp_t isn't exactly the same as cudaDeviceProp_t, because the former has both new and missing fields in the struct compared to the latter. Currently the code I wrote works on AMD GPUs only and segfaults when I try it on an NVIDIA GPU, which I believe is due to accessing fields that are nonexistent in cudaDeviceProp_t. It is also still missing a critical part to detect the exact GPU model within the same gfx??? GCN architecture code.

How do I figure out whether the detected GPU is AMD or NVIDIA?

Edit: for comparison, SYCL has sycl::info::device::vendor that provides this information.

Etienne M · Accepted Answer · 2023-02-14T14:04:14.767

3

When using HIP you known at compile time if you are compiling for AMD or Nvidia GPUs. There is no support for both AMD and Nvidia GPU code in the same translation unit.

Thus you could try relying on the following pre-processor definitions:

#if defined(__HIP_PLATFORM_AMD__)
    // AMD GPU code should take this code path
#elif defined(__HIP_PLATFORM_NVIDIA__)
    // Nvidia takes this code path.
#else
    #error "Unknown platform"
#endif

edited Feb 14 '23 at 14:04

answered Nov 08 '22 at 10:08

Etienne M

604
3
11

Finally, something that doesn't rely on anything else outside HIP! Thanks a lot! – wyphan Nov 08 '22 at 16:41
Also, I heard fat binary support will be coming in LLVM 15, so once ROCm/HIP picks it up, then it will be possible to have both AMD and NVIDIA GPU binaries in the same executable. – wyphan Nov 09 '22 at 16:07

score 0 · Answer 2 · answered Oct 16 '22 at 11:01

0

You can get the vendor ID with IDXGIAdapter::VendorId. If you need to get a vendor's ID, there are websites like PCI Lookup.

You will need a Direct3D and DXGI device.

#define VENDOR_NVIDIA 0x10DE

extern IDXGIAdapter* pDXGIAdapter;

std::string GetGPUVendor()
{
    DXGI_ADAPTER_DESC Description;
    if (SUCCEEDED(pDXGIAdapter->GetDesc(&Description)))
    {
        switch (Description.VendorId)
        {
            case VENDOR_NVIDIA:
                return "NVIDIA";
            // ...
            default:
                return "Unknown";
        }
    }
    else
    {
        return "Failed";
    }
}

answered Oct 16 '22 at 11:01

ShxShlix

11
2
3

Welp, unfortunately Direct3D is out of the question here, since I'm developing for Linux. – wyphan Oct 20 '22 at 18:29
1

@wyphan If you can use OpenGL, you can use glGetString(GL_VENDOR). It should return "NVIDIA Corporation" for NVIDIA, and "ATI Technologies Inc." for most AMD cards. – ShxShlix Oct 23 '22 at 16:25
I can use OpenGL, thanks! – wyphan Oct 26 '22 at 17:12

How to detect whether GPU is AMD or NVIDIA from inside HIP code

2 Answers2