• C
    Add Cuda event and stream API (#32460) · 9c7f6af5
    chentianyu03 提交于
    * add cuda event and stream api
    
    * add cuda event and stream api
    
    * add get_current_stream api
    
    * add get_current_stream api
    
    * init streams
    
    * modify get_current_stream
    
    * modify get_cuttent_stream
    
    * add synchronize func
    
    * add current_stream doc and test file
    
    * move get_current_stream into CUDA macro
    
    * move CudaEvent into CUDA macro
    
    * move _get_current_stream and _device_synchronize into cuda macro
    
    * modify the macro of cuda stream and event
    
    * add test case for synchronize
    
    * add paddle.devices.cuda module
    
    * event and stream support hip
    
    * add doc for stream and event class
    
    * move cuda stream and event into single pybind
    
    * add cuda_streams_py.cc to cmakelist
    
    * add _device_synchronize and _get_current_stream to core module
    
    * add test case for cudastream and cudaevent
    
    * move __all__ in streams.py
    
    * fix test fail
    
    * add cuda to devices __all__
    
    * fix current_stream doc writing error
    
    * move devices to device direction, and merge device.py into __init__.py
    
    * add required:gpu to sample codes
    
    * remove cuda direction from device/__init__.py
    9c7f6af5
pybind.cc 123.3 KB