HeterogeneousCore/ROCmUtilities/test/testRequireROCmDevices.cpp - master - githubmirror/cmssw

HeterogeneousCore/ROCmUtilities/test/testRequireROCmDevices.cpp

21 строка546 B

Andrea Bocci

Rename GPU check macros to uppercase

24 июл 2026, 18:36

24 июл 2026, 18:364ca8016

// Catch2 headers
#define CATCH_CONFIG_MAIN
#include <catch2/catch_all.hpp>

// ROCm headers
#include <hip/hip_runtime.h>

// CMSSW headers
#include "HeterogeneousCore/ROCmUtilities/interface/hipCheck.h"
#include "HeterogeneousCore/ROCmUtilities/interface/requireDevices.h"

TEST_CASE("HeterogeneousCore/ROCmUtilities testRequireROCmDevices", "[testRequireROCmDevices]") {
  SECTION("Test requireDevices()") {
    cms::rocmtest::requireDevices();

    int devices = 0;
    HIP_CHECK(hipGetDeviceCount(&devices));

    REQUIRE(devices > 0);
  }
}