Skip to content

Commit

Permalink
minor cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
gedoensmax committed Mar 20, 2023
1 parent f52a8d7 commit e61667f
Showing 1 changed file with 1 addition and 6 deletions.
7 changes: 1 addition & 6 deletions libvmaf/src/feature/cuda/integer_psnr_cuda.c
Original file line number Diff line number Diff line change
Expand Up @@ -231,13 +231,8 @@ static int extract_fex_cuda(VmafFeatureExtractor *fex,
CHECK_CUDA(cuEventRecord(s->event, vmaf_cuda_picture_get_stream(ref_pic)));
// This event ensures the input buffer is consumed
CHECK_CUDA(cuStreamWaitEvent(s->str, s->event, CU_EVENT_WAIT_DEFAULT));
// CHECK_CUDA(cuCtxPushCurrent(fex->cu_state->ctx));
// CHECK_CUDA(cuEventDestroy(s->event));
// CHECK_CUDA(cuEventCreate(&s->event, CU_EVENT_DEFAULT));
// CHECK_CUDA(cuCtxPopCurrent(NULL));

// Download sad
// CHECK_CUDA(cuStreamSynchronize(s->host_stream));
// Download SSE
CHECK_CUDA(cuMemcpyDtoHAsync(s->sse_host, (CUdeviceptr)s->sse_device->data,
sizeof(uint64_t) * 3, s->str));
CHECK_CUDA(cuEventRecord(s->finished, s->str));
Expand Down

0 comments on commit e61667f

Please sign in to comment.