• kaiJIN's avatar
    Support for running on arbitrary CUDA device. (#537) · f0318794
    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
README.md 11.9 KB