/
vuron
/
adept
Обзор
Документация
Войти
/
vuron
/
adept
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
src/backends/gpu/vulkan/vulkanbuffer.hpp
12 строк
447 B
kolkir
Vulkan tensor basic initialization
01 апр 2025, 23:01
01 апр 2025, 23:01
15de81e
Код
Авторство
О чём код?
#pragma once #include <adept/backends/buffer.h> #include <kompute/Kompute.hpp> namespace adept::gpu::vulkan { struct VulkanBuffer : public Buffer { VulkanBuffer(std::shared_ptr<kp::Tensor> kp_tensor) : kp_tensor(std::move(kp_tensor)) {} void* ptr() const override { return kp_tensor->rawData(); } size_t size() const override { return kp_tensor->memorySize(); } std::shared_ptr<kp::Tensor> kp_tensor; }; } // namespace adept::gpu::vulkan