-
kaiJIN authored
* support for any one cuda device * Revert "support for any one cuda device" This reverts commit 0197e4e2ef18ec41cc155f3ae2a0face5b77e1e9. * support runnning for anyone cuda device * using safe CUDAGuard rather than intrinsic CUDASetDevice * supplement a header dependency (test passed) * Support for arbitrary GPU device. * Support for arbitrary GPU device. * add docs for two method to control devices
f0318794
| Name |
Last commit
|
Last update |
|---|---|---|
| .. | ||
| cpu | ||
| cuda | ||
| ROIAlign.h | ||
| ROIPool.h | ||
| SigmoidFocalLoss.h | ||
| nms.h | ||
| vision.cpp |