@@ -149,6 +149,7 @@ typedef struct CudaFunctions {
|
149 | 149 | tcuDeviceGetPCIBusId *cuDeviceGetPCIBusId; |
150 | 150 | tcuDeviceComputeCapability *cuDeviceComputeCapability; |
151 | 151 | tcuCtxCreate_v2 *cuCtxCreate; |
| 152 | +tcuCtxGetCurrent *cuCtxGetCurrent; |
152 | 153 | tcuCtxSetLimit *cuCtxSetLimit; |
153 | 154 | tcuCtxPushCurrent_v2 *cuCtxPushCurrent; |
154 | 155 | tcuCtxPopCurrent_v2 *cuCtxPopCurrent; |
@@ -315,6 +316,7 @@ static inline int cuda_load_functions(CudaFunctions **functions, void *logctx)
|
315 | 316 | LOAD_SYMBOL(cuDeviceGetName, tcuDeviceGetName, "cuDeviceGetName"); |
316 | 317 | LOAD_SYMBOL(cuDeviceComputeCapability, tcuDeviceComputeCapability, "cuDeviceComputeCapability"); |
317 | 318 | LOAD_SYMBOL(cuCtxCreate, tcuCtxCreate_v2, "cuCtxCreate_v2"); |
| 319 | +LOAD_SYMBOL(cuCtxGetCurrent, tcuCtxGetCurrent, "cuCtxGetCurrent"); |
318 | 320 | LOAD_SYMBOL(cuCtxSetLimit, tcuCtxSetLimit, "cuCtxSetLimit"); |
319 | 321 | LOAD_SYMBOL(cuCtxPushCurrent, tcuCtxPushCurrent_v2, "cuCtxPushCurrent_v2"); |
320 | 322 | LOAD_SYMBOL(cuCtxPopCurrent, tcuCtxPopCurrent_v2, "cuCtxPopCurrent_v2"); |
|