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/Event.h (487B)

#ifndef THCP_EVENT_INC #define THCP_EVENT_INC #include #include #include // NOLINTNEXTLINE(cppcoreguidelines-pro-type-member-init) struct THCPEvent { PyObject_HEAD at::cuda::CUDAEvent cuda_event; }; extern PyObject *THCPEventClass; void THCPEvent_init(PyObject *module); inline bool THCPEvent_Check(PyObject* obj) { return THCPEventClass && PyObject_IsInstance(obj, THCPEventClass); } #endif // THCP_EVENT_INC