Commit 3fb4bd2e by liyinqiao

Bug fixed.

Fix the bug in destroy the stream of XDevice.
parent b4785f07
...@@ -147,12 +147,6 @@ XTensor::XTensor(const int myOrder, const int* myDimSize, const TENSOR_DATA_TYPE ...@@ -147,12 +147,6 @@ XTensor::XTensor(const int myOrder, const int* myDimSize, const TENSOR_DATA_TYPE
if(order >= 0) if(order >= 0)
Resize(myOrder, myDimSize, myDataType, myDenseRatio); Resize(myOrder, myDimSize, myDataType, myDenseRatio);
#ifdef USE_CUDA
int gpuNum = 0;
cudaError_t error = cudaGetDeviceCount(&gpuNum);
#endif
atexit(XDevManager::DelDeviceStream);
} }
/* copy constructor */ /* copy constructor */
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论