OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
22,007 Commits 8,304 Branches 150 Tags 1.5 GiB
aa30176c68
Commit Graph

3 Commits

Author SHA1 Message Date
Ilia Cherniavskii
19956b200d Relax set_num_threads restriction in parallel native case (#27947) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/27947

Don't throw exception if the requested size is the same as the currently
used one

Test Plan:
ATEN_THREADING=NATIVE python setup.py develop --cmake

Imported from OSS

Differential Revision: D17919416

fbshipit-source-id: 411f7c9bd6a46e7a003b43a200c2ce3b76453a2e
 2019-10-16 21:53:36 -07:00
Ilia Cherniavskii
5835165ce3 Add get/set_num_interop_threads into torch.h include (#20659) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/20659
ghimport-source-id: 4858d03a9f89c613f64901c3430a7b212f76eb95

Reviewed By: dzhulgakov

Differential Revision: D15399780

Pulled By: ilia-cher

fbshipit-source-id: c1c3cb628c5ee664468f9d181bcd76a5105a89fd
 2019-05-20 00:34:59 -07:00
Will Feng
0087069dce Use torch::get/set_num_threads without additional includes beyond torch/torch.h (#20176) 
Summary:
Fixes https://github.com/pytorch/pytorch/issues/20130.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/20176

Differential Revision: D15275036

Pulled By: yf225

fbshipit-source-id: 0f04e1fbfed18c07030b20e92e957ef5f2b5707d
 2019-05-09 06:08:27 -07:00