libvmaf_cuda.h

Go to the documentation of this file.

 
#ifndef __VMAF_CUDA_H__
#define __VMAF_CUDA_H__
 
#include "libvmaf/libvmaf.h"
 
#ifdef __cplusplus
extern "C" {
#endif
 
typedef struct VmafCudaState VmafCudaState;
 
typedef struct VmafCudaConfiguration {
    void *cu_ctx; 
} VmafCudaConfiguration;

int vmaf_cuda_state_init(VmafCudaState **cu_state, VmafCudaConfiguration cfg);

int vmaf_cuda_import_state(VmafContext *vmaf, VmafCudaState *cu_state);
 
enum VmafCudaPicturePreallocationMethod {
    VMAF_CUDA_PICTURE_PREALLOCATION_METHOD_NONE = 0,
    VMAF_CUDA_PICTURE_PREALLOCATION_METHOD_DEVICE,
    VMAF_CUDA_PICTURE_PREALLOCATION_METHOD_HOST,
    VMAF_CUDA_PICTURE_PREALLOCATION_METHOD_HOST_PINNED,
};
 
typedef struct VmafCudaPictureConfiguration {
    struct {
        unsigned w, h;
        unsigned bpc;
        enum VmafPixelFormat pix_fmt;
    } pic_params;
    enum VmafCudaPicturePreallocationMethod pic_prealloc_method;
} VmafCudaPictureConfiguration;

int vmaf_cuda_preallocate_pictures(VmafContext *vmaf,
                                   VmafCudaPictureConfiguration cfg);

int vmaf_cuda_fetch_preallocated_picture(VmafContext *vmaf, VmafPicture* pic);
 
#ifdef __cplusplus
}
#endif
 
#endif /* __VMAF_CUDA_H__ */