dump DXGI & NVML PCI bus information

nvmldxgi.cpp

#include <windows.h>
#include <dxgi.h>
#include <D3dkmthk.h>
#include <stdio.h>

#include "nvml\include\nvml.h"

#pragma comment(lib, "nvml\\lib\\nvml.lib")
#pragma comment(lib, "dxgi.lib")

int main()
{
	IDXGIFactory* factory;
	CreateDXGIFactory(IID_PPV_ARGS(&factory));

	UINT i = 0;
	do
	{
		IDXGIAdapter* adapter;
		factory->EnumAdapters(i++, &adapter);
		if (!adapter)
			break;

		DXGI_ADAPTER_DESC desc;
		adapter->GetDesc(&desc);
		adapter->Release();

		wprintf(L"adapter #%u LUID 0x%08X.0x%08X: %s\n", i, desc.AdapterLuid.HighPart, desc.AdapterLuid.HighPart, desc.Description);

		D3DKMT_OPENADAPTERFROMLUID luid{};
		luid.AdapterLuid = desc.AdapterLuid;
		D3DKMTOpenAdapterFromLuid(&luid);
		wprintf(L" D3D handle 0x%08X\n", luid.hAdapter);

		D3DKMT_ADAPTERADDRESS address{};
		D3DKMT_QUERYADAPTERINFO info{};
		info.hAdapter = luid.hAdapter;
		info.Type = KMTQAITYPE_ADAPTERADDRESS;
		info.PrivateDriverDataSize = sizeof(address);
		info.pPrivateDriverData = &address;
		D3DKMTQueryAdapterInfo(&info);
		wprintf(L" D3D bus 0x%08X device 0x%08X function 0x%08X\n", address.BusNumber, address.DeviceNumber, address.FunctionNumber);

		D3DKMT_ADAPTERTYPE rinfo{};
		info.Type = KMTQAITYPE_ADAPTERTYPE;
		info.PrivateDriverDataSize = sizeof(rinfo);
		info.pPrivateDriverData = &rinfo;
		D3DKMTQueryAdapterInfo(&info);
		wprintf(L"\n");

	} while (true);
	factory->Release();

	nvmlInit_v2();
	UINT count = 0;
	nvmlDeviceGetCount_v2(&count);

	for (UINT i = 0; i < count; i++)
	{
		nvmlDevice_t device;
		nvmlDeviceGetHandleByIndex(i, &device);

		char name[NVML_DEVICE_NAME_BUFFER_SIZE];
		nvmlDeviceGetName(device, name, NVML_DEVICE_NAME_BUFFER_SIZE);

		nvmlPciInfo_t pci;
		nvmlDeviceGetPciInfo(device, &pci);

		printf("NVML device %u domain.bus.device.function %s 0x%08X subSystem 0x%08X %s\n", i, pci.busId, pci.pciDeviceId, pci.pciSubSystemId, name);
	}
	nvmlShutdown();
}