HeterogeneousCore/CUDAUtilities/test/testRequireCUDADevices.cpp - master - githubmirror/cmssw

HeterogeneousCore/CUDAUtilities/test/testRequireCUDADevices.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>

// CUDA headers
#include <cuda_runtime.h>

// CMSSW headers
#include "HeterogeneousCore/CUDAUtilities/interface/cudaCheck.h"
#include "HeterogeneousCore/CUDAUtilities/interface/requireDevices.h"

TEST_CASE("HeterogeneousCore/CUDAUtilities testRequireCUDADevices", "[testRequireCUDADevices]") {
  SECTION("Test requireDevices()") {
    cms::cudatest::requireDevices();

    int devices = 0;
    CUDA_CHECK(cudaGetDeviceCount(&devices));

    REQUIRE(devices > 0);
  }
}