pytorch

OSSForks/pytorch

Fork 0

mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00

Commit Graph

Author	SHA1	Message	Date
Zachary DeVito	d985cf46f1	Add workaround to fix include warnings in Python 2 builds. (#6716 )	2018-04-24 12:30:19 -07:00
Sam Gross	bcfe259f83	Add streams and comms as optional arguments (#3968 ) Adds streams and comms as optional arguments to the NCCL calls in torch.cuda.nccl. Also exposes ncclUniqueId and ncclCommInitRank for multi-process mode. Moves Py_RETURN_NONE statements after the GIL is re-acquired.	2017-12-04 13:51:22 -05:00
Sam Gross	b43ce05268	Refactor parts of utils.h (#648 ) Moves THPObjectPtr into a separate header, so that it can be included independently. Currently, utils.h requries all of THP.h. Also adds RAII structs for acquiring and releasing the GIL.	2017-01-30 21:16:28 -05:00

Author

SHA1

Message

Date

Zachary DeVito

d985cf46f1

Add workaround to fix include warnings in Python 2 builds. (#6716 )

2018-04-24 12:30:19 -07:00

Sam Gross

bcfe259f83

Add streams and comms as optional arguments (#3968 )

Adds streams and comms as optional arguments to the NCCL calls in
torch.cuda.nccl. Also exposes ncclUniqueId and ncclCommInitRank for
multi-process mode.

Moves Py_RETURN_NONE statements after the GIL is re-acquired.

2017-12-04 13:51:22 -05:00

Sam Gross

b43ce05268

Refactor parts of utils.h (#648 )

Moves THPObjectPtr into a separate header, so that it can be included
independently. Currently, utils.h requries all of THP.h. Also adds RAII
structs for acquiring and releasing the GIL.

2017-01-30 21:16:28 -05:00

3 Commits