Commit 2db2d137 authored by Tomoaki Teshima's avatar Tomoaki Teshima

follow the naming rule as other API

  * stop using 'CUDA' prefix
  * add explanation
parent b2698f24
...@@ -855,7 +855,14 @@ private: ...@@ -855,7 +855,14 @@ private:
CV_EXPORTS void printCudaDeviceInfo(int device); CV_EXPORTS void printCudaDeviceInfo(int device);
CV_EXPORTS void printShortCudaDeviceInfo(int device); CV_EXPORTS void printShortCudaDeviceInfo(int device);
CV_EXPORTS void convertFp16Cuda(InputArray _src, OutputArray _dst, Stream& stream = Stream::Null()); /** @brief Converts an array to half precision floating number.
@param _src input array.
@param _dst output array.
@param stream Stream for the asynchronous version.
@sa convertFp16
*/
CV_EXPORTS void convertFp16(InputArray _src, OutputArray _dst, Stream& stream = Stream::Null());
//! @} cudacore_init //! @} cudacore_init
......
...@@ -594,7 +594,7 @@ void cv::cuda::GpuMat::convertTo(OutputArray _dst, int rtype, double alpha, doub ...@@ -594,7 +594,7 @@ void cv::cuda::GpuMat::convertTo(OutputArray _dst, int rtype, double alpha, doub
funcs[sdepth][ddepth](reshape(1), dst.reshape(1), alpha, beta, stream); funcs[sdepth][ddepth](reshape(1), dst.reshape(1), alpha, beta, stream);
} }
void cv::cuda::convertFp16Cuda(InputArray _src, OutputArray _dst, Stream& stream) void cv::cuda::convertFp16(InputArray _src, OutputArray _dst, Stream& stream)
{ {
GpuMat src = _src.getGpuMat(); GpuMat src = _src.getGpuMat();
int ddepth = 0; int ddepth = 0;
......
...@@ -101,11 +101,11 @@ public: ...@@ -101,11 +101,11 @@ public:
GpuMat g_dst; GpuMat g_dst;
// Fp32 -> Fp16 // Fp32 -> Fp16
convertFp16Cuda(g_src, g_dst); cuda::convertFp16(g_src, g_dst);
convertFp16Cuda(g_dst.clone(), g_dst); cv::convertFp16(src, dst);
// Fp16 -> Fp32 // Fp16 -> Fp32
convertFp16(src, dst); cuda::convertFp16(g_dst.clone(), g_dst);
convertFp16(dst, ref); cv::convertFp16(dst, ref);
g_dst.download(dst); g_dst.download(dst);
EXPECT_MAT_NEAR(dst, ref, 0.0); EXPECT_MAT_NEAR(dst, ref, 0.0);
...@@ -127,8 +127,8 @@ public: ...@@ -127,8 +127,8 @@ public:
GpuMat g_dst; GpuMat g_dst;
// Fp32 -> Fp16 // Fp32 -> Fp16
convertFp16Cuda(g_src, g_dst); cuda::convertFp16(g_src, g_dst);
convertFp16(src, ref); cv::convertFp16(src, ref);
g_dst.download(dst); g_dst.download(dst);
EXPECT_MAT_NEAR(dst, ref, 0.0); EXPECT_MAT_NEAR(dst, ref, 0.0);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment