Commit Graph

1407 Commits

Author SHA1 Message Date
Adam Paszke
c81491b37d Preserve directory structure when installing headers 2017-01-31 01:58:09 +01:00
Adam Paszke
42e189425f Detect ZMQ libs and headers in CMake 2017-01-31 01:58:09 +01:00
Adam Paszke
3cfa0d7199 Expose C API for process group mode 2017-01-31 01:58:09 +01:00
Adam Paszke
7c9e088661 Reorganize THD directory structure 2017-01-31 01:58:09 +01:00
Mateusz Piotrowski
e78aa4bb84 Implement CommandChannel with ZMQ. 2017-01-31 01:58:09 +01:00
Janusz Marcinkiewicz
f8e94d0d8b Implement DataChannel (MPI and TCP) (#8) 2017-01-31 01:58:09 +01:00
Filip Binkiewicz
ebe6f40fce RPC message packing and unpacking implemented 2017-01-31 01:58:09 +01:00
Adam Paszke
5fb37efb46 Use #pragma once instead of defines 2017-01-31 01:58:09 +01:00
Adam Paszke
4f47855873 Style improvements 2017-01-31 01:58:09 +01:00
Adam Paszke
52ae6f682f Add initial version of tensor wrappers 2017-01-31 01:58:09 +01:00
Adam Paszke
c35f58f97b Template for THD implementation 2017-01-31 01:58:09 +01:00
Adam Paszke
e24eee04f0 Link THC to THPP 2017-01-28 23:37:39 -05:00
Sam Gross
27c4c6e0af Merge commit '6ee77b4edd1552d3a9a2e5389ffc351e513a8089' 2017-01-27 17:29:07 -08:00
Sam Gross
da17414b3f Merge commit '343d65db91c2419843d36aed5467c2d1374108bc' 2017-01-27 17:16:08 -08:00
Sam Gross
be2b27a747 Merge commit '4461ae809043390d5223905cb82b17035c7f9f31' 2017-01-27 17:15:21 -08:00
Sam Gross
aec2c8f752 Merge commit 'c45ff2efe64d0face3889194ba6f885fe9cc4d48' 2017-01-27 17:12:13 -08:00
Sam Gross
887002e932 Add bindings to CUDA tensors and storages in THPP (#615) 2017-01-27 18:15:56 -05:00
Soumith Chintala
a2938e3d11 add cc 3.0 to nccl (#594) 2017-01-25 22:47:23 -05:00
Luke Yeager
138ee75a3b Fix for target_link_libraries on CMake 2.8 (#581) 2017-01-24 17:26:24 -05:00
soumith
116e0c7f38 Merge commit '45596d52897fb187701943cb77456ff1e7249989' 2017-01-23 14:37:44 -08:00
Soumith Chintala
342e7b873d fixing THPP cmake for cmake < 3.1 (#559) 2017-01-23 14:47:06 -05:00
Sam Gross
c9db9c2317 Add C++ tensor library (from THD fork) (#526) 2017-01-20 15:23:34 -05:00
soumith
205b9bc05f fix build_all.sh 2017-01-17 16:55:46 -08:00
soumith
3d40c0562d improve build_all.sh 2017-01-17 09:49:48 -08:00
soumith
146bcc0e70 adding binary build copy option to build_all 2017-01-17 07:52:18 -08:00
Natalia Gimelshein
2290798a83 if nccl is available, do not compile it and load system version 2017-01-14 10:09:48 +01:00
soumith
fd600b11a6 Merge commit '2b88d85505d7317f980e69201e72694d6d5905a4' 2017-01-13 15:58:54 -08:00
soumith
b5c9f5c4c3 Merge commit 'ca74bb17b8823d74b83433e2743f23e572501c72' 2017-01-13 15:55:19 -08:00
soumith
b8a5b1ed8e Merge commit 'e67b525388a5ae11ed243e94bbc25b4934b03a66' 2017-01-13 15:54:49 -08:00
soumith
e8102b0a9b fix compiler warning in THCS 2017-01-09 15:19:13 -08:00
Zeming Lin
59d66e6963 Sparse Library (#333) 2017-01-05 00:43:41 +01:00
soumith
01ac2d3791 Merge commit '1b97f088cb9e42717122795463a800bf3f503adf' 2017-01-02 09:39:45 -08:00
soumith
eac687df5a Merge commit '849cbf3a4774727eadb97c27af13bfbdc976a02a' 2017-01-02 09:39:20 -08:00
soumith
7ee152881e Merge commit '3074f8eb8103ecdcbbcbb8d49332d9e7d6f3141c' 2017-01-01 01:13:17 -08:00
soumith
748208775f Merge commit '5df17050bf82337d13dbd2108bd17922ac38956c' 2017-01-01 00:08:55 -08:00
soumith
92df0eb2bf removing unneeded flags in build_all.sh 2016-12-31 20:16:50 -08:00
soumith
995195935b Merge commit 'be8376eb883d2f5a466994e024cde44e6adc6130' 2016-12-31 20:10:11 -08:00
Soumith Chintala
b5dc36f278 explicitly linking against v1 libs to avoid lua-torch conflicts (#386) 2016-12-31 10:30:36 -05:00
soumith
41976e2b60 Merge commit '3dac1b9936a62225cf8516d6d7830fe6c83039ae' 2016-12-30 21:07:13 -08:00
soumith
d2bb56647f Merge commit '224422eed6813c15b3c3b2c0dcd5e0187ec660a1' 2016-12-30 19:51:01 -08:00
soumith
3c26f7a205 Merge commit '10f78985e72fb6834b435ac3f8d0890fa6614365' 2016-12-30 19:24:00 -08:00
soumith
9ac9809f27 Merge commit 'd8f4d5f91e3680478a6843d49d7295c1165618f0' 2016-12-30 19:23:41 -08:00
soumith
7bf6e984ef Merge commit 'dc95f66a954ad18b80f3f649f8e2c8507c048b74' 2016-12-30 19:23:17 -08:00
soumith
47f56f0230 Merge commit '43fbdd3b45d4351623a4aa9c8d5e6dba9eac259a' 2016-12-30 17:46:04 -08:00
soumith
b4018c4c30 Merge commit '803d0320771365754658ac74587cc082c2a61fa7' 2016-12-30 17:45:45 -08:00
soumith
9d2d884313 Merge commit 'b5cf1d2fc71604f472a07d0181a05a7f09e276c2' 2016-12-30 16:50:25 -08:00
soumith
c0600e655a Merge commit 'c1ca9044bd6dccd293471c6caeeeea4ebd97d61b' 2016-12-30 16:49:56 -08:00
soumith
671ed89f2a Merge commit '52c2a92013c45afa5df61a68b16695663ee9fab5' 2016-12-30 16:49:29 -08:00
soumith
e0372643e1 Merge commit '541ab961d8f9a02bbbe1a06ba25027116ee93c20' 2016-12-30 16:49:05 -08:00
soumith
491d5ba4fd add new flags to build_all.sh 2016-12-29 18:16:59 -08:00
soumith
d42eadfeb9 Merge commit '2975f539ff8ac9b8e07fb2b610bd69a1596d4c3c' 2016-12-29 17:51:34 -08:00
soumith
9a40821069 Merge commit '1ac038ab243bb2718b37cbd81eadbfeb2a234252' 2016-12-29 17:51:13 -08:00
Sam Gross
24af02154c Use ForkingPickler for sharing tensor/storages across processes (#344)
This hooks into the (internal) ForkingPickler class in multiprocessing
to reduce tensors, storages, and CUDA events instead of our queue from
joblib. This makes it easier to use the standard multiprocessing classes
in later versions of Python.

This also exposes:

 - Tensor/Storage.share_memory_()
 - Module.share_memory()

These methods move the CPU tensors and storages to shared memory. If
you're using the "fork" method of multiprocessing, these objects can be
directly inherited instead of serialized through a queue.
2016-12-28 20:34:23 -05:00
Sam Gross
29918c6ca5 Copy libnccl.so.1 instead of libnccl.so
Occasionally, my PyTorch checkout gets into a bad state where libnccl.so
does not exist, but the NCCL makefile doesn't build it because
libnccl.so.1 exists. Switch to copying libnccl.so.1 to work around this.
2016-12-28 20:21:31 +01:00
soumith
179d5efc81 Merge commit '310ec57fd7176e07137ab7bc717f3602b6f53aa5' 2016-12-28 07:33:37 -08:00
soumith
e704ec5c6f Merge commit '46f024846698cd8201d6c1804f21bffda15a2069' 2016-12-27 19:12:45 -08:00
soumith
6cda6bb34c Merge commit 'd2a93c310292c9427056e02ac7e0d5cca12a04a2' 2016-12-27 19:12:21 -08:00
soumith
41f9c14297 Merge commit '135687f04a4e4e0722c14f096c9a1fc647c95f07' 2016-12-27 13:12:26 -08:00
soumith
b03407289f Merge commit '55a794e6ec8d01fc8cceee14ce23ec501e517570' 2016-12-24 11:06:27 -08:00
soumith
10faa303bc Merge commit '6fa371cb0db9f43e3d05746c7e90516975052589' 2016-12-22 17:35:13 -08:00
soumith
800e24616a Merge commit 'fa61159dd0bfd9bbb190e1dfbd90a68f4d3c30c8' 2016-12-21 12:40:41 -08:00
soumith
d63a435787 Merge commit 'f16a624b35dd28fbd4cdcd3bd08dfc2421c3e2b0' 2016-12-21 12:40:20 -08:00
soumith
cc24b68584 Merge commit 'f413ee087df1a4bbd8b5a9baba83d07ae0729ea0' 2016-12-16 05:29:16 -08:00
soumith
287ba38905 Merge commit 'ed9dbff4e0295dbeb2e8de908cb8c1109c278a8a' 2016-12-13 17:23:56 -08:00
soumith
6ba4e48521 Merge commit '3adcb2c157ed7df5aaff9b59d4526aa24ec770db' 2016-12-13 16:49:38 -08:00
soumith
b7269f2295 Merge commit '220183ed783101f19d88cb8fb3052fd4abc7234f' 2016-12-13 16:49:15 -08:00
soumith
5ab317d4a6 Merge commit '258c9ffb2c2d23a06b153aa9161a88ad930cfbbc' 2016-12-13 16:48:45 -08:00
soumith
431bcf7afa Merge commit '56245426ebcf239363867905ca2a4cea676dd45d' 2016-12-13 16:48:16 -08:00
Sergey Zagoruyko
04393cd47d fix gcc-6 build on os x (#297) 2016-12-12 00:01:15 +01:00
Sam Gross
09c94a170c Merge commit 'f2a18004a77f146bb5b431715402f4afd3cacccd' 2016-12-01 22:16:58 -08:00
Sam Gross
bf0e185bd6 Merge commit 'bb1019d1ec1503718b97d17366902f96f349f472' 2016-12-01 13:47:20 -08:00
Sam Gross
5b3ccec10d Merge commit 'c2d32030a25e352eb2e2af26931163c0f4c96b36' 2016-12-01 13:46:35 -08:00
Sam Gross
eb07581502 Merge commit 'bec6ab47b6782f60925e306b69e0f556274fb28e' 2016-12-01 13:46:03 -08:00
Sam Gross
934a2b6878 Merge commit 'b27d4de850b5f43829bd4980f5e7f3b4b32ab7cf' 2016-12-01 13:45:05 -08:00
soumith
61063ebade Merge commit 'a7f24ccb7635447b133011d39e36279be140149e' 2016-11-26 09:13:12 -08:00
soumith
3e70e26278 Merge commit '08a1bc71c0712a4151de83d1487a55b218ae1a15' 2016-11-26 09:12:53 -08:00
soumith
66e7e42800 Merge commit '379860e457dbb72c0f18e0366e5b199452b302f5' 2016-11-26 09:12:24 -08:00
soumith
a9bd27ce5c Merge commit '709255d9952783eed6c8f84e504693f9b436f852' 2016-11-22 13:26:09 -08:00
soumith
2e36c4ea2d Merge commit 'f3cb636294fbd0e15dd4b3bfdca16e73d1dca38b' 2016-11-22 13:25:53 -08:00
soumith
4e45385a8d Merge commit 'b27f576f29189ca78dd670cbd177bfa29b695c50' 2016-11-22 13:25:29 -08:00
soumith
cf5e925c10 Merge commit 'f6b94dd830c06692cd78addd41868a7a12c48755' 2016-11-22 13:25:00 -08:00
Adam Paszke
3911a1d395 Fix memory leak in LogSoftMax 2016-11-21 21:32:10 +01:00
Gregory Chanan
ebd3648fd6 Call newContiguous rather than arg checking isContiguous. 2016-11-21 21:32:10 +01:00
Gregory Chanan
f698f09cb7 Add contiguous checking / make tensors contiguous for
SpatialUpSamplingBilinear, PReLU, SpatialSubSampling, TemporalConvolution.
2016-11-21 21:32:10 +01:00
Gregory Chanan
86aa5dae05 Move VolumetricConvolution contiguous code from lua to C. 2016-11-21 21:32:10 +01:00
Adam Paszke
ae6f2dd11c Adapt nn code to changes in THNN and THCUNN 2016-11-15 23:02:14 +01:00
soumith
456998f043 Merge commit 'aeed8a6ea4650d1092289a60e71d8d83875a0ba6' 2016-11-15 12:55:11 -08:00
soumith
c09f07edd9 Merge commit 'c82537462baa715b2c70726f7da8f734b2ad3a3f' 2016-11-15 12:53:29 -08:00
soumith
c279a91c03 Merge commit '64c8a1377335799b322ca41d323dee13118be0ab' 2016-11-13 21:54:27 -08:00
soumith
ef6a764509 Merge commit '1cee5a359c2828800db0c41ebe0108bd5eef9501' 2016-11-13 15:23:11 -08:00
soumith
4db5afdf7e Merge commit 'f2daa616d105d700b63f05c4d544befb6e65a036' 2016-11-13 15:20:03 -08:00
soumith
7867187451 Merge commit '4f8e6ec42abd5b9b5491a49bdfe1a637e6675207' 2016-11-13 15:19:10 -08:00
Adam Paszke
bc08011e72 Don't jongjmp out of omp loops in unpooling modules 2016-11-08 18:12:56 +01:00
Gregory Chanan
7cccc216d0 ArgCheck that dilation parameters are > 0. 2016-11-08 18:12:56 +01:00
soumith
293bfb03dd Merge commit '4def4e696b9079f587d0dba3e86423df5ea429b8' 2016-11-03 14:12:22 -07:00
soumith
bf00308ab2 Merge commit 'fd677945741b4ee353079911993ada3770e07f5c' 2016-11-03 13:31:12 -07:00
Sam Gross
0676cad200 Merge commit 'e644f6ed2c1965b0de55cc9037d5c75245f63d54' 2016-11-03 08:36:42 -07:00
Sam Gross
3b1d217310 Merge commit 'e32af0196e10ad11b3938ad73ec5ef49cac7c03e' 2016-11-03 08:36:04 -07:00
soumith
7a06dbb87e Merge commit '1234e434fa2b6ddd440194c8bccd352593902c69' 2016-11-01 21:33:41 -07:00
Soumith Chintala
2d374f982e Changes for ccache nvcc support 2016-11-01 15:54:33 -04:00
soumith
0dfec752a3 Merge commit 'f16f68e103dfc22921f6106ec7136ddc7a0ab087' 2016-11-01 10:38:13 -07:00
Brandon Amos
6df334ea68 Improve potrf error message. (#189) 2016-10-31 18:48:29 -04:00
Sam Gross
2ee451f5f7 Build in Release mode 2016-10-28 12:51:19 -07:00
Sam Gross
2af3098e5a Merge commit '42e835ebb81a3ecf8f76e15bb1866c1427f61d74' 2016-10-27 13:49:23 -07:00
Sam Gross
2e44511b13 Merge commit 'bbe8627a3f0e6cbb8fd1952826f75df741e44b01' 2016-10-27 13:47:36 -07:00
Sam Gross
7bc4aa7e72 Merge commit '2bd36604e298547cc66f175588c925271223b4e9' 2016-10-27 13:46:38 -07:00
Adam Paszke
ffd4863b23 Don't build nccl on macOS 2016-10-24 22:29:43 +02:00
Sam Gross
d293c17d21 Merge commit '1a3920e5dc546803ec8ada369ff1b0d56cf24e76' 2016-10-17 10:29:41 -07:00
Sam Gross
7d10b2370f Merge commit 'ec7a2878013ec70a4d4a8bfb6f5e5503f87f9ea0' 2016-10-17 09:35:04 -07:00
Sam Gross
31ec7650ac Merge commit '429f2d67652f4fcba0bbf65c7d3e109e136a9cdf' 2016-10-17 09:33:06 -07:00
Sam Gross
c014920dc1 Merge commit 'b01c78580594c53e6afb02b3d2110577a4673308' 2016-10-17 09:32:01 -07:00
Sam Gross
17e3d4e1ee Merge commit '38cb3d02270b9e558a891a9a2bef01a75d1bd9e1' 2016-10-17 09:31:38 -07:00
Sam Gross
c15648c6b5 Add NCCL build scripts 2016-10-14 14:16:32 -07:00
soumith
b34654bf97 Merge commit 'ab0e86ae4b0a08b8d0a67f1494ff80e65a6932ad' 2016-10-02 20:58:29 -07:00
soumith
6068df3ab2 Merge commit '60a8a9e918e04fd5581d20e4e7527dd115c69cd8' 2016-10-02 20:56:33 -07:00
soumith
bb35999f51 Merge commit '25c51c49aa3bb9ac5f64560a46f1f2a905f4e3f7' 2016-10-02 20:55:38 -07:00
Sam Gross
ff59385034 Add 'torch/lib/nccl/' from commit 'ca330b110ae76ace344182ab83a028911111cc36'
git-subtree-dir: torch/lib/nccl
git-subtree-mainline: ea4f812a12
git-subtree-split: ca330b110a
2016-09-30 16:35:16 -07:00
Adam Paszke
c1c0969834 Allow changing the default error handler for all threads
THSetErrorHandler still modifies per-thread pointers, but
THSetDefaultErrorHandler allows to set a handler that's
used by all threads that haven't specified any function.
2016-09-30 14:59:50 -07:00
BTNC
b87f26ce26 windows high resolution timer with a few makefile changes (#776)
windows high resolution timer
2016-09-30 14:59:50 -07:00
Rui Guo
67335e638c bug fix for read/writeLong in THMemoryFile 2016-09-30 14:59:50 -07:00
Rui Guo
90916f34a7 fix cpuid ecx; change to compile with msvc 2016-09-30 14:59:50 -07:00
Sam Gross
3e1c88e3e0 Merge commit 'da1e3f084d237ba319a22987f95f70abb69d7745' 2016-09-30 11:07:46 -07:00
Sam Gross
e98a4ea336 Merge commit '0b0a62420c52b6e4d4c80c36d067db4654d1ed8d' 2016-09-30 11:06:53 -07:00
Adam Paszke
941cf4e63d Add ffi utils for user C extensions 2016-09-29 09:35:56 -07:00
soumith
4a5557203b Merge commit 'c020a8502bd943aa37f897efe79a01fd61249ab4' 2016-09-26 17:54:05 -07:00
Sam Gross
fa1f286cae Merge commit '85bd287b7ba481312fa58d7ffb32cba901c58829' 2016-09-26 13:08:32 -07:00
Soumith Chintala
1cf87e8a0b OSX + Python 2 build fixes 2016-09-25 19:26:13 -04:00
Adam Paszke
e223564a55 Fix multiprocessing on OS X 2016-09-16 18:27:07 -04:00
Adam Paszke
6584b35db2 Add getDevice for CUDA storages 2016-09-15 18:48:27 -07:00
Adam Paszke
7920b9229b Update TH 2016-09-15 12:47:53 -07:00
Adam Paszke
60f4d285af Update THCUNN 2016-09-14 11:08:37 -04:00
Adam Paszke
aa092f03d8 Update THNN 2016-09-14 11:08:25 -04:00
soumith
1d9b10d312 libshm needs static libstdc++ on binary build 2016-09-13 13:13:58 -07:00
soumith
26a614b4d1 Merge commit '788ff68c1f273185092710abb269fe550f0fe196' 2016-09-13 11:21:04 -07:00
soumith
297fa957f7 Merge commit 'a442b5f5cc0e250653688bab8b4be93bfd3934ed' 2016-09-13 11:17:04 -07:00
soumith
05fb544f23 Merge commit '73d15cf64320b4b77e7393efa1bf1e913404cfd6' 2016-09-13 11:16:09 -07:00
soumith
c65e795435 Merge commit 'eb6419f02bea8bca3a8ff1791d0a9f2a2e733035' 2016-09-13 10:33:43 -07:00
Adam Paszke
58f507f9e3 Add file descriptor sharing mode to multiprocessing 2016-09-08 11:23:33 -07:00
Adam Paszke
46fa7d987b Update TH 2016-09-08 09:54:21 -07:00
Soumith Chintala
4cffa2219a build fixes for OSX 2016-09-06 22:06:06 -04:00
Adam Paszke
f9d186d33a Add initial version of multiprocessing module 2016-08-31 19:46:08 -07:00
Adam Paszke
c7a66ddf74 Add new shared memory allocator to TH 2016-08-31 19:28:43 -07:00
Soumith Chintala
b60d7e1476 fix addmm and addmv documentation when they are methods 2016-08-31 19:28:43 -07:00
Adam Paszke
686e8d32e2 Add torch.save and torch.load 2016-08-23 07:51:55 -07:00
Adam Paszke
8d933cbfc4 Fixes for OS X 2016-08-22 22:45:35 -04:00
Adam Paszke
aadac369dc Update THCUNN 2016-08-19 14:19:27 -07:00
Adam Paszke
62dcb1aed0 Update THNN 2016-08-19 14:19:10 -07:00
Adam Paszke
d8a5882890 Update THCUNN 2016-08-12 21:52:20 -07:00
Adam Paszke
5bf8aa2b49 Update THC 2016-08-12 21:52:01 -07:00
Adam Paszke
73c6312391 Update THNN 2016-08-12 18:33:49 -07:00
Adam Paszke
928ca7bfb0 Update TH 2016-08-12 18:31:03 -07:00
Adam Paszke
e9f9fd3727 Major refactor 2016-08-10 09:24:53 -07:00
Adam Paszke
652a31b714 Add build scripts for libraries 2016-08-04 14:12:31 -07:00
Adam Paszke
5dc6a67bdc Add 'torch/lib/THCUNN/' from commit '1395c27906a1e93a9e5e8ba5bb0e85d35f6838b3'
git-subtree-dir: torch/lib/THCUNN
git-subtree-mainline: 286bd362c4
git-subtree-split: 1395c27906
2016-08-04 11:35:56 -07:00
Adam Paszke
286bd362c4 Add 'torch/lib/THC/' from commit 'b85e5e6df4a3a46d656d9b01e3245e44631326c7'
git-subtree-dir: torch/lib/THC
git-subtree-mainline: 035eb28e18
git-subtree-split: b85e5e6df4
2016-08-04 11:35:43 -07:00
Adam Paszke
035eb28e18 Add 'torch/lib/THNN/' from commit '4fe7059a315d156ecd080ff7bd5b4fe3d3a9efad'
git-subtree-dir: torch/lib/THNN
git-subtree-mainline: c3f0c1e2e0
git-subtree-split: 4fe7059a31
2016-08-04 10:58:50 -07:00
Adam Paszke
c3f0c1e2e0 Add 'torch/lib/TH/' from commit '9f108b5d548dd286f114f7a4a02fc33f4cf40be5'
git-subtree-dir: torch/lib/TH
git-subtree-mainline: d7504b1f52
git-subtree-split: 9f108b5d54
2016-08-04 10:58:45 -07:00