Enable Embree build with SYCL support and OneAPI 2024.1.0

3rdparty/embree/embree.cmake

@@ -64,6 +64,14 @@ else()
 endif()
 
 
+if(BUILD_SYCL_MODULE)
+    set(ISA_ARGS ${ISA_ARGS} -DCMAKE_CXX_COMPILER=icpx)
+    set(ISA_ARGS ${ISA_ARGS} -DCMAKE_C_COMPILER=icx)
+    set(ISA_ARGS ${ISA_ARGS} -DEMBREE_SYCL_SUPPORT=ON)
+    #list(APPEND ISA_LIBS embree4_sycl)
+endif()
+
+
 ExternalProject_Add(
     ext_embree
     PREFIX embree
@@ -92,10 +100,12 @@ ExternalProject_Add(
         <INSTALL_DIR>/${Open3D_INSTALL_LIB_DIR}/${CMAKE_STATIC_LIBRARY_PREFIX}sys${CMAKE_STATIC_LIBRARY_SUFFIX}
         <INSTALL_DIR>/${Open3D_INSTALL_LIB_DIR}/${CMAKE_STATIC_LIBRARY_PREFIX}math${CMAKE_STATIC_LIBRARY_SUFFIX}
         <INSTALL_DIR>/${Open3D_INSTALL_LIB_DIR}/${CMAKE_STATIC_LIBRARY_PREFIX}tasking${CMAKE_STATIC_LIBRARY_SUFFIX}
+        <INSTALL_DIR>/${Open3D_INSTALL_LIB_DIR}/${CMAKE_STATIC_LIBRARY_PREFIX}ze_wrapper${CMAKE_STATIC_LIBRARY_SUFFIX}
+        <INSTALL_DIR>/${Open3D_INSTALL_LIB_DIR}/${CMAKE_STATIC_LIBRARY_PREFIX}embree_rthwif${CMAKE_STATIC_LIBRARY_SUFFIX}
         ${ISA_BUILD_BYPRODUCTS}
 )
 
 ExternalProject_Get_Property(ext_embree INSTALL_DIR)
 set(EMBREE_INCLUDE_DIRS ${INSTALL_DIR}/include/ ${INSTALL_DIR}/src/ext_embree/) # "/" is critical.
 set(EMBREE_LIB_DIR ${INSTALL_DIR}/${Open3D_INSTALL_LIB_DIR})
-set(EMBREE_LIBRARIES embree4 ${ISA_LIBS} simd lexers sys math tasking)
+set(EMBREE_LIBRARIES embree4_sycl embree4 simd lexers sys math tasking ze_wrapper embree_rthwif ${ISA_LIBS})

cpp/open3d/core/Device.cpp

@@ -137,7 +137,7 @@ std::vector<Device> Device::GetAvailableCUDADevices() {
 }
 
 std::vector<Device> Device::GetAvailableSYCLDevices() {
-    return sycl::GetAvailableSYCLDevices();
+    return sy::GetAvailableSYCLDevices();
 }
 
 void Device::PrintAvailableDevices() {

cpp/open3d/core/MemoryManagerSYCL.cpp

@@ -5,8 +5,8 @@
 // SPDX-License-Identifier: MIT
 // ----------------------------------------------------------------------------
 
-#include <CL/sycl.hpp>
 #include <cstdlib>
+#include <sycl/sycl.hpp>
 #include <unordered_map>
 
 #include "open3d/core/Device.h"
@@ -18,24 +18,22 @@
 namespace open3d {
 namespace core {
 
-namespace sy = cl::sycl;
-
 void* MemoryManagerSYCL::Malloc(size_t byte_size, const Device& device) {
-    const sy::queue& queue =
-            sycl::SYCLContext::GetInstance().GetDefaultQueue(device);
+    const sycl::queue& queue =
+            sy::SYCLContext::GetInstance().GetDefaultQueue(device);
 
 #ifdef ENABLE_SYCL_UNIFIED_SHARED_MEMORY
-    return static_cast<void*>(sy::malloc_shared(byte_size, queue));
+    return static_cast<void*>(sycl::malloc_shared(byte_size, queue));
 #else
-    return static_cast<void*>(sy::malloc_device(byte_size, queue));
+    return static_cast<void*>(sycl::malloc_device(byte_size, queue));
 #endif
 }
 
 void MemoryManagerSYCL::Free(void* ptr, const Device& device) {
     if (ptr) {
-        const sy::queue& queue =
-                sycl::SYCLContext::GetInstance().GetDefaultQueue(device);
-        sy::free(ptr, queue);
+        const sycl::queue& queue =
+                sy::SYCLContext::GetInstance().GetDefaultQueue(device);
+        sycl::free(ptr, queue);
     }
 }
 
@@ -62,8 +60,8 @@ void MemoryManagerSYCL::Memcpy(void* dst_ptr,
                           dst_device.ToString());
     }
 
-    sy::queue queue =
-            sycl::SYCLContext::GetInstance().GetDefaultQueue(device_with_queue);
+    sycl::queue queue =
+            sy::SYCLContext::GetInstance().GetDefaultQueue(device_with_queue);
     queue.memcpy(dst_ptr, src_ptr, num_bytes).wait_and_throw();
 }
 

cpp/open3d/core/SYCLContext.cpp

@@ -7,17 +7,17 @@
 
 #include "open3d/core/SYCLContext.h"
 
-#include <CL/sycl.hpp>
 #include <array>
 #include <cstdlib>
 #include <sstream>
+#include <sycl/sycl.hpp>
 
 #include "open3d/core/SYCLUtils.h"
 #include "open3d/utility/Logging.h"
 
 namespace open3d {
 namespace core {
-namespace sycl {
+namespace sy {
 
 SYCLContext &SYCLContext::GetInstance() {
     static thread_local SYCLContext instance;
@@ -38,36 +38,37 @@ bool SYCLContext::IsDeviceAvailable(const Device &device) {
 }
 std::vector<Device> SYCLContext::GetAvailableSYCLDevices() { return devices_; }
 
-sy::queue SYCLContext::GetDefaultQueue(const Device &device) {
+sycl::queue SYCLContext::GetDefaultQueue(const Device &device) {
     return device_to_default_queue_.at(device);
 }
 
 SYCLContext::SYCLContext() {
     // SYCL GPU.
     // TODO: Currently we only support one GPU device.
     try {
-        const sy::device &sycl_device = sy::device(sy::gpu_selector());
+        const sycl::device &sycl_device = sycl::device(sycl::gpu_selector());
         const Device open3d_device = Device("SYCL:0");
         devices_.push_back(open3d_device);
         device_to_sycl_device_[open3d_device] = sycl_device;
-        device_to_default_queue_[open3d_device] = sy::queue(sycl_device);
-    } catch (const sy::exception &e) {
+        device_to_default_queue_[open3d_device] = sycl::queue(sycl_device);
+    } catch (const sycl::exception &e) {
     }
 
     if (devices_.size() == 0) {
         // SYCL CPU fallback.
         // This could happen if the Intel GPGPU driver is not installed or if
         // your CPU does not have integrated GPU.
         try {
-            const sy::device &sycl_device = sy::device(sy::host_selector());
+            const sycl::device &sycl_device =
+                    sycl::device(sycl::host_selector());
             const Device open3d_device = Device("SYCL:0");
             utility::LogWarning(
                     "SYCL GPU device is not available, falling back to SYCL "
                     "host device.");
             devices_.push_back(open3d_device);
             device_to_sycl_device_[open3d_device] = sycl_device;
-            device_to_default_queue_[open3d_device] = sy::queue(sycl_device);
-        } catch (const sy::exception &e) {
+            device_to_default_queue_[open3d_device] = sycl::queue(sycl_device);
+        } catch (const sycl::exception &e) {
         }
     }
 
@@ -76,6 +77,6 @@ SYCLContext::SYCLContext() {
     }
 }
 
-}  // namespace sycl
+}  // namespace sy
 }  // namespace core
 }  // namespace open3d

cpp/open3d/core/SYCLContext.h

@@ -14,16 +14,14 @@
 
 #pragma once
 
-#include <CL/sycl.hpp>
+#include <sycl/sycl.hpp>
 #include <unordered_map>
 
 #include "open3d/core/Device.h"
 
 namespace open3d {
 namespace core {
-namespace sycl {
-
-namespace sy = cl::sycl;
+namespace sy {
 
 /// Singleton SYCL context manager. It maintains:
 /// - A default queue for each SYCL device
@@ -45,7 +43,7 @@ class SYCLContext {
     std::vector<Device> GetAvailableSYCLDevices();
 
     /// Get the default SYCL queue given an Open3D device.
-    sy::queue GetDefaultQueue(const Device& device);
+    sycl::queue GetDefaultQueue(const Device& device);
 
 private:
     SYCLContext();
@@ -54,12 +52,12 @@ class SYCLContext {
     std::vector<Device> devices_;
 
     /// Maps core::Device to the corresponding default SYCL queue.
-    std::unordered_map<Device, sy::queue> device_to_default_queue_;
+    std::unordered_map<Device, sycl::queue> device_to_default_queue_;
 
-    /// Maps core::Device to sy::device. Internal use only for now.
-    std::unordered_map<Device, sy::device> device_to_sycl_device_;
+    /// Maps core::Device to sycl::device. Internal use only for now.
+    std::unordered_map<Device, sycl::device> device_to_sycl_device_;
 };
 
-}  // namespace sycl
+}  // namespace sy
 }  // namespace core
 }  // namespace open3d