c++ - 根据模板 C++ 的类型具有可变类型的类成员

Question

我正在尝试编写一个模板类来管理主机和 Nvidia GPU 上的指针，类似于 boost::shared_ptr .为此，我需要在类中定义 2 个指针，一个用于主机，一个用于 GPU。

template<typename T>
class GPUSharedPtr
{
  protected:

  T *cpu_pointer;
  T *gpu_pointer;
}

CUDA 的问题在于它为 std::complex<double> 定义了自己的类型和 std::complex<float> , 这是 cuDoubleComplex和 cuComplex分别。

template<typename std::complex<double> >
class GPUSharedPtr
{
  protected:

  std::complex<double> *cpu_pointer;
  cuDoubleComplex *gpu_pointer;
}


template<typename std::complex<float> >
class GPUSharedPtr
{
  protected:

  std::complex<float> *cpu_pointer;
  cuComplex *gpu_pointer;
}

在这两种情况下，如何为 GPU 使用不同类型的指针？

谢谢!

Answer 1

为什么不将两者都添加为模板参数:

template<typename CPU_T, typename GPU_T>
class GPUSharedPtr
{
  protected:
      CPU_T *cpu_pointer;
      GPU_T *gpu_pointer;
};

int main()
{
    GPUSharedPtr<std::complex<double> , cuDoubleComplex> g1;
    GPUSharedPtr<std::complex<float> , cuComplex> g2;
    return 0;
}

或者(如果在您的情况下更有意义)，您可以按照 MooingDuck 的建议使用标签分派(dispatch)来正确默认 GPU 指针类型的值:

template<class CPU_T> struct DefaultGPU_T { typedef CPU_T type;};
template<> struct DefaultGPU_T<std::complex<double>> { typedef cuDoubleComplex type;};
template<> struct DefaultGPU_T<std::complex<float>> { typedef cuComplex type;};

template<typename CPU_T, typename GPU_T=typename DefaultGPU_T<CPU_T>::type>

class GPUSharedPtr
{
protected:
    CPU_T *cpu_pointer;
    GPU_T *gpu_pointer;
};

int main()
{
    GPUSharedPtr<int> g0;
    GPUSharedPtr<std::complex<double>> g1;
    GPUSharedPtr<std::complex<float>> g2;
    return 0;
}

---

Live demo.