Commit 5546a260 by xiaotong

formatting

parent d8bd1ef2
...@@ -35,12 +35,13 @@ void _ConvertDataType(const XTensor * input, XTensor * output) ...@@ -35,12 +35,13 @@ void _ConvertDataType(const XTensor * input, XTensor * output)
CheckNTErrors((input->unitSize == output->unitSize), "Input and Output must be same in size!"); CheckNTErrors((input->unitSize == output->unitSize), "Input and Output must be same in size!");
if (input->dataType == output->dataType) if (input->dataType == output->dataType)
return; return;
#ifdef USE_CUDA #ifdef USE_CUDA
/* run it on GPUs */ /* run it on GPUs */
if (input->devID >= 0) { if (input->devID >= 0) {
_CudaConvertDataType(input, output); _CudaConvertDataType(input, output);
return; return;
} }
#endif #endif
if (input->dataType == X_FLOAT && output->dataType == X_INT) { if (input->dataType == X_FLOAT && output->dataType == X_INT) {
...@@ -59,4 +60,4 @@ void _ConvertDataType(const XTensor * input, XTensor * output) ...@@ -59,4 +60,4 @@ void _ConvertDataType(const XTensor * input, XTensor * output)
ShowNTErrors("Unsupported data types for conversion!"); ShowNTErrors("Unsupported data types for conversion!");
} }
} // namespace nts(NiuTrans.Tensor) } // namespace nts(NiuTrans.Tensor)
\ No newline at end of file
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论