Commit 5546a260 by xiaotong

formatting

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