Commit 666ce55
committed
Fix initialization of CUDA context on the default device during pipeline creation (#829)
- when DALI pipeline is created there was CUDA call without preceding setDevice
what caused the creation of a context on the default device (GPU0). Add DeviceGuard.
Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>1 parent 8a8bfcd commit 666ce55
1 file changed
+2
-0
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
325 | 325 | | |
326 | 326 | | |
327 | 327 | | |
| 328 | + | |
| 329 | + | |
328 | 330 | | |
329 | 331 | | |
330 | 332 | | |
| |||
0 commit comments