diff options
| author | Yangqing Jia <jiayq84@gmail.com> | 2016-08-04 15:17:15 -0700 |
|---|---|---|
| committer | Yangqing Jia <jiayq84@gmail.com> | 2016-08-04 15:17:15 -0700 |
| commit | 2362571b230ddae2c362958209bde77aaceed6f7 (patch) | |
| tree | e07436249de8fa2cf38655029e2e7b2e9626fbc2 | |
| parent | 1ede7a7ff0fc163dbe54f138bb2287ff9cea6beb (diff) | |
| download | pytorch-2362571b230ddae2c362958209bde77aaceed6f7.tar.gz pytorch-2362571b230ddae2c362958209bde77aaceed6f7.tar.bz2 pytorch-2362571b230ddae2c362958209bde77aaceed6f7.zip | |
add submodule cub
| -rw-r--r-- | .gitmodules | 3 | ||||
| -rw-r--r-- | third_party/BREW | 10 | ||||
| m--------- | third_party/cub | 0 |
3 files changed, 13 insertions, 0 deletions
diff --git a/.gitmodules b/.gitmodules index d41d8385db..d9d9a81a19 100644 --- a/.gitmodules +++ b/.gitmodules @@ -7,3 +7,6 @@ [submodule "third_party/cnmem"] path = third_party/cnmem url = https://github.com/nvidia/cnmem.git +[submodule "third_party/cub"] + path = third_party/cub + url = https://github.com/NVlabs/cub.git diff --git a/third_party/BREW b/third_party/BREW index 7f4ea5a9d0..47e829bbab 100644 --- a/third_party/BREW +++ b/third_party/BREW @@ -137,6 +137,16 @@ cuda_library( ], ) +shell_script( + name = "cub", + srcs = Glob(["cub/cub/*.h", "cub/cub/*/*.h", "cub/cub/*/*/*.h"]), + commands = [ + "DST=$CAFFE2_GENDIR/third_party/include/", + "mkdir -p $DST", + "cp -r $CAFFE2_SRCDIR/$CAFFE2_CWD/cub/cub $DST/", + ], +) + ############################################################################### # Other libraries # This section includes all other libraries in alphabet order. diff --git a/third_party/cub b/third_party/cub new file mode 160000 +Subproject c8c00d5700053c173b9dbf324c0ea3aaafec747 |