函數模板打印推力矢量

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, " "));