scottk/UltraGrid
1
0
Fork 0
mirror of https://github.com/outbackdingo/UltraGrid.git synced 2026-04-07 16:05:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

vcomp/cmpto_j2k: fixed setting cuda_device!=0

Browse Source
This commit is contained in:
Martin Pulec
2024-08-29 12:00:59 +02:00
parent ca71f59b9c
commit 4061f8d0f2
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/video_compress/cmpto_j2k.cpp
View File

@@ -108,6 +108,9 @@ struct cmpto_j2k_enc_cuda_buffer_data_allocator
: public video_frame_pool_allocator {
void *allocate(size_t size) override
{
if (alloc == cuda_wrapper_malloc) {
cuda_wrapper_set_device((int) cuda_devices[0]);
}
void *ptr = nullptr;
if (CUDA_WRAPPER_SUCCESS !=
alloc(&ptr, size)) {
@@ -255,6 +258,7 @@ static shared_ptr<video_frame> get_copy(struct state_video_compress_j2k *s, vide
s->convertFunc(ret.get(), frame);
} else if (s->pool_in_device_memory) {
#ifdef HAVE_CUDA
cuda_wrapper_set_device((int) cuda_devices[0]);
cuda_wrapper_memcpy(ret->tiles[0].data, frame->tiles[0].data,
frame->tiles[0].data_len,
CUDA_WRAPPER_MEMCPY_HOST_TO_DEVICE);