cuda/utility/type_conversion.hpp

0001 /*
0002  * SPDX-PackageName: "covfie, a part of the ACTS project"
0003  * SPDX-FileCopyrightText: 2022 CERN
0004  * SPDX-License-Identifier: MPL-2.0
0005  */
0006
0007 #pragma once
0008
0009 #include <memory>
0010
0011 #include <cuda_runtime.h>
0012
0013 #include <covfie/core/vector.hpp>
0014
0015 namespace covfie::utility {
0016 template <typename T>
0017 struct to_cuda_channel_t {
0018 };
0019
0020 template <>
0021 struct to_cuda_channel_t<covfie::vector::float1> {
0022     using type = float;
0023 };
0024
0025 template <>
0026 struct to_cuda_channel_t<covfie::vector::float2> {
0027     using type = ::float2;
0028 };
0029
0030 template <>
0031 struct to_cuda_channel_t<covfie::vector::float3> {
0032     using type = ::float4;
0033 };
0034
0035 template <>
0036 struct to_cuda_channel_t<covfie::vector::float4> {
0037     using type = ::float4;
0038 };
0039 }