mirror of https://github.com/davisking/dlib.git
removed print statements
This commit is contained in:
parent
adc791eadf
commit
694902920d
|
@ -22,7 +22,6 @@ namespace dlib
|
||||||
{
|
{
|
||||||
if (have_active_transfer)
|
if (have_active_transfer)
|
||||||
{
|
{
|
||||||
std::cout << "wait for cudaStreamSynchronize()" << std::endl;
|
|
||||||
CHECK_CUDA(cudaStreamSynchronize((cudaStream_t)cuda_stream.get()));
|
CHECK_CUDA(cudaStreamSynchronize((cudaStream_t)cuda_stream.get()));
|
||||||
have_active_transfer = false;
|
have_active_transfer = false;
|
||||||
// Check for errors. These calls to cudaGetLastError() are what help us find
|
// Check for errors. These calls to cudaGetLastError() are what help us find
|
||||||
|
@ -46,7 +45,6 @@ namespace dlib
|
||||||
if (!host_current)
|
if (!host_current)
|
||||||
{
|
{
|
||||||
wait_for_transfer_to_finish();
|
wait_for_transfer_to_finish();
|
||||||
std::cout << "cudaMemcpy to host" << std::endl;
|
|
||||||
CHECK_CUDA(cudaMemcpy(data_host.get(), data_device.get(), data_size*sizeof(float), cudaMemcpyDeviceToHost));
|
CHECK_CUDA(cudaMemcpy(data_host.get(), data_device.get(), data_size*sizeof(float), cudaMemcpyDeviceToHost));
|
||||||
host_current = true;
|
host_current = true;
|
||||||
// Check for errors. These calls to cudaGetLastError() are what help us find
|
// Check for errors. These calls to cudaGetLastError() are what help us find
|
||||||
|
@ -60,7 +58,6 @@ namespace dlib
|
||||||
{
|
{
|
||||||
if (!device_current)
|
if (!device_current)
|
||||||
{
|
{
|
||||||
std::cout << "cudaMemcpyAsync to device" << std::endl;
|
|
||||||
CHECK_CUDA(cudaMemcpyAsync(data_device.get(), data_host.get(), data_size*sizeof(float), cudaMemcpyHostToDevice, (cudaStream_t)cuda_stream.get()));
|
CHECK_CUDA(cudaMemcpyAsync(data_device.get(), data_host.get(), data_size*sizeof(float), cudaMemcpyHostToDevice, (cudaStream_t)cuda_stream.get()));
|
||||||
have_active_transfer = true;
|
have_active_transfer = true;
|
||||||
device_current = true;
|
device_current = true;
|
||||||
|
|
Loading…
Reference in New Issue