ComputeGPUVulkan.h

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        Open3D: www.open3d.org                            -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2024 www.open3d.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
//
// Vulkan compute backend for the Gaussian splatting pipeline.
// Compiled on non-Apple platforms.
//
// Key design:
//   - Implements GaussianSplatGpuContext backed by Vulkan compute pipelines
//     loaded from SPIR-V assets at resources/gaussian_splat/.
//   - Uses VK_KHR_push_descriptor for efficient per-dispatch binding (no
//     descriptor pool allocation per frame).
//   - Uses VMA for general buffer/image allocation.  Shared cross-API images
//     (color, depth) are registered via RegisterSharedImageInVulkanContext()
//     rather than allocated here; VulkanInteropContext owns those.
//   - Synchronisation: each EndXxxPass() submits and waits (fence-based) so
//     the rest of the pipeline sees a completed GPU result.
 
#pragma once
 
#if !defined(__APPLE__)
 
#include <cstdint>
#include <memory>
 
#include "open3d/visualization/rendering/gaussian_splat/ComputeGPU.h"
 
namespace open3d {
namespace visualization {
namespace rendering {
 
void RegisterSharedImageInVulkanContext(GaussianSplatGpuContext& ctx,
                                        std::uint32_t gl_name,
                                        std::uintptr_t vk_image_opaque,
                                        std::uint32_t vk_format_opaque,
                                        std::uint32_t width,
                                        std::uint32_t height);
 
void UnregisterSharedImageFromVulkanContext(GaussianSplatGpuContext& ctx,
                                            std::uint32_t gl_name);
 
[[nodiscard]] std::unique_ptr<GaussianSplatGpuContext>
CreateComputeGpuContextVulkan();
 
}  // namespace rendering
}  // namespace visualization
}  // namespace open3d
 
#endif  // !defined(__APPLE__)