SYCLContext.h

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        Open3D: www.open3d.org                            -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2024 www.open3d.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
 
 
#pragma once
 
#include <map>
#include <sycl/sycl.hpp>
 
#include "open3d/core/Device.h"
 
namespace open3d {
namespace core {
namespace sy {
 
struct SYCLDevice {
    SYCLDevice(const sycl::device& sycl_device);
    std::string name;            
    std::string device_type;     
    sycl::device device;         
    sycl::queue queue;           
    size_t max_work_group_size;  
    bool fp64;  
    bool usm_device_allocations;  
};
 
class SYCLContext {
public:
    SYCLContext(SYCLContext const&) = delete;
    void operator=(SYCLContext const&) = delete;
 
    static SYCLContext& GetInstance();
 
    bool IsAvailable();
 
    bool IsDeviceAvailable(const Device& device);
 
    std::vector<Device> GetAvailableSYCLDevices();
 
    sycl::queue GetDefaultQueue(const Device& device);
 
    SYCLDevice GetDeviceProperties(const Device& device) {
        return devices_.at(device);
    };
 
private:
    SYCLContext();
 
    std::map<Device, SYCLDevice> devices_;
};
 
}  // namespace sy
}  // namespace core
}  // namespace open3d