函数模板打印推力矢量

Question

我正在写一个matrix模板类，打印到文件都和std::cout，即：

matrix<float> myMat; 
... 
myMat.cout(...)   // print to std::cout 
myMat.write("out.txt") // print to file 

双方将分享其中我想实现一个共同的基本打印功能作为模板，因为我看到了使用thrust::copy将数据写入std::cout和文件的不同示例。

下面是我所做的骨架，但它目前正在输出垃圾。任何人都可以指出我可能做出的一些错误？例如，我是否允许像这样通过std::cout？

template <typename data_T> matrix { 
    ... 

    template <typename out_T> int printTo(out_T &out, ...) { 
     data_T *start = ..., *end = ...; 
     ... 
     thrust::copy(start, end, std::ostream_iterator<data_T>(out, " ")); 
     ... 
    } 

    int cout(...) { 
     ... 
     printTo(std::cout, ...); 
     ... 
    } 

    int write(char* path, ...) { 
     ... 
     std::ofstream file; 
     file.open(path); 
     printTo(file, ...); 
     ... 
    } 
} 

编辑：

• 更改为int printTo(std::ostream &out, ...) {...}不能解决问题。
• 更多信息：我从thrust::device_vector<T>数据读入矩阵，说dvec，并将其转换为使用thrust::raw_pointer_cast(&dvec[0])一个data_T指针pvec（作为CUBLAS库使用原始指针）。然后我运行pvec，然后打印出来。
• 我试过直接从原始thrust::device_vector的指针打印（即*dvec），它确实工作：thrust::copy((*dvec).begin(), (*dvec).begin() + n ...)。那么，为什么我只能使用*dvec迭代器而不是原始指针pvec来复制？

Answer 1

请不要在此处使用raw_pointer_cast。这会欺骗推测你已经有了一个指向主机数据的指针，这就是为什么你的代码不能给你所期望的。我会期望你的代码只是崩溃。

要将device_vector复制到ostream_iterator，只需直接使用thrust::copy：

thrust::device_vector<float> vec = ... 

thrust::copy(vec.begin(), vec.end(), std::ostream_iterator<float>(std::cout, " "));