@@ -203,7 +203,7 @@ class C10_CUDA_API CUDAStream {
203203 * isHighPriority to true, or a stream for a specific device by setting device
204204 * (defaulting to the current CUDA stream.)
205205 */
206- TORCH_API CUDAStream
206+ C10_API CUDAStream
207207getStreamFromPool (const bool isHighPriority = false , DeviceIndex device = -1 );
208208
209209/* *
@@ -213,7 +213,7 @@ getStreamFromPool(const bool isHighPriority = false, DeviceIndex device = -1);
213213 * want to operate on a non-torch allocated stream for data exchange or similar
214214 * purposes
215215 */
216- TORCH_API CUDAStream
216+ C10_API CUDAStream
217217getStreamFromExternal (cudaStream_t ext_stream, DeviceIndex device_index);
218218
219219/* *
@@ -222,7 +222,7 @@ getStreamFromExternal(cudaStream_t ext_stream, DeviceIndex device_index);
222222 * where most computation occurs when you aren't explicitly using
223223 * streams.
224224 */
225- TORCH_API CUDAStream getDefaultCUDAStream (DeviceIndex device_index = -1 );
225+ C10_API CUDAStream getDefaultCUDAStream (DeviceIndex device_index = -1 );
226226
227227/* *
228228 * Get the current CUDA stream, for the passed CUDA device, or for the
@@ -231,7 +231,7 @@ TORCH_API CUDAStream getDefaultCUDAStream(DeviceIndex device_index = -1);
231231 * be different if someone called 'setCurrentCUDAStream' or used 'StreamGuard'
232232 * or 'CUDAStreamGuard'.
233233 */
234- TORCH_API CUDAStream getCurrentCUDAStream (DeviceIndex device_index = -1 );
234+ C10_API CUDAStream getCurrentCUDAStream (DeviceIndex device_index = -1 );
235235
236236/* *
237237 * Set the current stream on the device of the passed in stream to be
@@ -243,7 +243,7 @@ TORCH_API CUDAStream getCurrentCUDAStream(DeviceIndex device_index = -1);
243243 * (which will switch both your current device and current stream in the way you
244244 * expect, and reset it back to its original state afterwards).
245245 */
246- TORCH_API void setCurrentCUDAStream (CUDAStream stream);
246+ C10_API void setCurrentCUDAStream (CUDAStream stream);
247247
248248C10_API std::ostream& operator <<(std::ostream& stream, const CUDAStream& s);
249249
0 commit comments