FM / cuda

/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/cuda/

/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/cuda

Name	Size	Mode	Actions
comm.h	1525	0644	edit dl rm
device_set.h	152	0644	edit dl rm
Event.h	487	0644	edit dl rm
Module.h	503	0644	edit dl rm
nccl.h	5037	0644	edit dl rm
override_macros.h	1887	0644	edit dl rm
python_comm.h	114	0644	edit dl rm
python_nccl.h	616	0644	edit dl rm
restore_macros.h	820	0644	edit dl rm
serialization.h	524	0644	edit dl rm
Storage.h	2585	0644	edit dl rm
Stream.h	524	0644	edit dl rm
THCP.h	421	0644	edit dl rm
undef_macros.h	932	0644	edit dl rm
utils.h	871	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/cuda/Stream.h (524B)

#ifndef THCP_STREAM_INC #define THCP_STREAM_INC #include #include #include #include // NOLINTNEXTLINE(cppcoreguidelines-pro-type-member-init) struct THCPStream : THPStream{ at::cuda::CUDAStream cuda_stream; }; extern PyObject *THCPStreamClass; void THCPStream_init(PyObject *module); inline bool THCPStream_Check(PyObject* obj) { return THCPStreamClass && PyObject_IsInstance(obj, THCPStreamClass); } #endif // THCP_STREAM_INC