PlatformGpuSyclNvidia.hpp

Go to the documentation of this file.

/* Copyright 2025 Aurora Perego
 * SPDX-License-Identifier: MPL-2.0
 */
 
#pragma once
 
#include "alpaka/dev/DevGenericSycl.hpp"
#include "alpaka/dev/Traits.hpp"
#include "alpaka/platform/PlatformGenericSycl.hpp"
 
#if defined(ALPAKA_ACC_SYCL_ENABLED) && defined(ALPAKA_SYCL_ONEAPI_GPU_NVIDIA)
 
#    include <sycl/sycl.hpp>
 
namespace alpaka
{
    namespace detail
    {
        template<>
        struct SYCLDeviceSelector<TagGpuSyclNvidia>
        {
            auto operator()(sycl::device const& dev) const -> int
            {
                auto const& vendor = dev.get_info<sycl::info::device::vendor>();
                auto const is_intel_gpu = dev.is_gpu() && (vendor.find("NVIDIA") != std::string::npos);
 
                return is_intel_gpu ? 1 : -1;
            }
        };
    } // namespace detail
 
    //! The SYCL device manager.
    using PlatformGpuSyclNvidia = PlatformGenericSycl<TagGpuSyclNvidia>;
} // namespace alpaka
 
#endif