eyalroz / cuda-api-wrappers

Thin, unified, C++-flavored wrappers for the CUDA APIs
BSD 3-Clause "New" or "Revised" License
766 stars 79 forks source link

Accept execution affinity parameters when creating a context #616

Open eyalroz opened 5 months ago

eyalroz commented 5 months ago

Beginning with CUDA 11.4, a new API call for creating contexts has become available:

CUresult cuCtxCreate_v3 ( 
    CUcontext* pctx,
    CUexecAffinityParam* paramsArray,
    int  numParams,
    unsigned int  flags,
    CUdevice dev )

we should support creating contexts with these execution parameters (and representing such affinities).