Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
515 commits
Select commit Hold shift + click to select a range
990835f
Merge pull request #2083 from jeffdonahue/tile-layer
jeffdonahue Aug 26, 2015
1f3f952
MVNLayer fixes.
jyegerlehner Aug 23, 2015
c548a79
Draw Deconvolution layers like Convolution layers
maaskola Aug 2, 2015
93b4835
Merge pull request #2981 from maaskola/draw-deconvolution
jeffdonahue Aug 26, 2015
215bea0
Merge pull request #2964 from jyegerlehner/mvn-layer-fixes
jeffdonahue Aug 26, 2015
292dbc5
Fix previous mistake on unimplemented top and address pyramid_height_…
ronghanghu Aug 24, 2015
b7e4bfe
Merge pull request #2970 from ronghanghu/spp-fix
ronghanghu Aug 26, 2015
f572eef
Merge pull request #2944 from philkr/python_layer_param
longjon Aug 27, 2015
4f64b9e
Add extra openblas search path
mattdawkins Aug 27, 2015
4d7fe4d
Fix EmbedLayer compiler warning for unused variable.
jyegerlehner Aug 27, 2015
4c561fd
Merge pull request #2989 from jyegerlehner/embed-layer-compile-warning
jeffdonahue Aug 27, 2015
846f2c3
fix GPU data race
longjon Aug 29, 2015
d3dc7b4
Merge pull request #2894 from mfigurnov/fix-truncation-warning
ronghanghu Aug 30, 2015
d362894
Merge pull request #2998 from longjon/data-race
jeffdonahue Aug 30, 2015
a675172
Compute backward for negative lr_mult
philkr Sep 1, 2015
2d55e31
Merge pull request #3007 from philkr/neg_lr_mult
jeffdonahue Sep 1, 2015
b04000e
Cleanup: Fixup capitalisation of Caffe_POSTFIX.
darrengarvey Sep 1, 2015
e8f96f5
Fix memory leak in convert_mnist_siamese_data.
darrengarvey Aug 10, 2015
6567ce1
Merge pull request #3008 from darrengarvey/cleanup-cmake-variable
jeffdonahue Sep 1, 2015
583194a
cifar10: Fix examples by setting snapshot_format.
darrengarvey Sep 1, 2015
6f5812c
Fix up documentation errors.
darrengarvey Sep 1, 2015
6ca0ab6
Show output from convert_imageset tool
lukeyeager Sep 2, 2015
e7a1dd0
Merge pull request #3009 from darrengarvey/cleanup-documentation-warn…
jeffdonahue Sep 2, 2015
d900435
Merge pull request #3011 from darrengarvey/fixup-cifar10-example
jeffdonahue Sep 2, 2015
35980eb
Merge pull request #3012 from lukeyeager/convert-imageset-logging
jeffdonahue Sep 2, 2015
cbb2ed1
Merge pull request #3010 from darrengarvey/fixup-memory-leak-siamese
jeffdonahue Sep 2, 2015
e8e660d
Merge pull request #2990 from mattdawkins/add-openblas-path
jeffdonahue Sep 2, 2015
d1a0345
SliceLayer: allow trivial operation with single top Blob
jeffdonahue Oct 7, 2014
784dfdd
ConcatLayer: allow trivial operation with single bottom Blob
jeffdonahue Jan 23, 2015
66823b5
Merge pull request #3014 from jeffdonahue/trivial-slice-concat
ronghanghu Sep 3, 2015
46c3f00
net.cpp fix debug_info params -> learnable_params
jeffdonahue Sep 3, 2015
c248474
NetSpec: don't require lists to specify single-element repeated fields
jeffdonahue Aug 22, 2015
50cbf01
Merge pull request #2959 from jeffdonahue/netspec-allow-non-iterable-…
jeffdonahue Sep 3, 2015
0486127
Merge pull request #3023 from jeffdonahue/param-debug-info-fix
shelhamer Sep 4, 2015
1bdc18c
Update extract_features.cpp
cdluminate Sep 4, 2015
aa40ab9
Fix AccuracyLayerTest for per-class accuracy.
ronghanghu Sep 4, 2015
c09c30d
Merge pull request #3027 from ronghanghu/fix-accuracy-test
shelhamer Sep 4, 2015
9a244f9
Merge pull request #3026 from CDLuminate/patch-1
jeffdonahue Sep 4, 2015
5cc76ad
enabling the alternate solvers to be accessed by the python interface
Sep 4, 2015
5367a1a
Merge pull request #3024 from danielgordon10/python-solver-fix
shelhamer Sep 4, 2015
1394cdc
disallow PythonLayer in Multi-GPU training
ronghanghu Sep 4, 2015
8fbac04
Minor: missing space in string formatting
Sep 9, 2015
4f8775c
Merge pull request #3048 from seanbell/minor-string-fix
ronghanghu Sep 9, 2015
3456259
Use EXPECT_NEAR in EltwiseLayer test
ronghanghu Sep 14, 2015
5b3ad4d
Merge pull request #2941 from thatguymike/MultiGPUDocs
ronghanghu Sep 14, 2015
f87658a
Merge pull request #3063 from ronghanghu/use-expectnear-in-eltwiselay…
ronghanghu Sep 14, 2015
ab554cb
Check that the snapshot directory is writeable before starting training
Sep 9, 2015
e4baef2
Merge pull request #3049 from seanbell/check-snapshot-prefix
ronghanghu Sep 14, 2015
b7f9cba
removed bug in caffe.io.resize_image when applied to Nd images
mohomran Sep 15, 2015
71e0587
Merge pull request #3070 from mohomran/resize_nd-image_bug
jeffdonahue Sep 16, 2015
3d3a8b2
Get back 'USE CPU' print for caffe train
ronghanghu Sep 16, 2015
f3a933a
Separate IO dependencies
eelstork Jun 27, 2015
2349c6d
Fix case in CMake notices
eelstork Sep 17, 2015
68c9e2b
Add a comment indicating that Travis CI tests are CPU only
eelstork Sep 8, 2015
5ff3734
Merge pull request #2523 from BonsaiAI/separate-io-dependencies
jeffdonahue Sep 17, 2015
2a585f7
Merge pull request #3074 from ronghanghu/show-use-cpu
ronghanghu Sep 17, 2015
e75ae96
[build] include IO dependencies by default
shelhamer Sep 17, 2015
3d12b5d
Merge pull request #3081 from shelhamer/include-io-by-default
shelhamer Sep 17, 2015
f75d594
refine format of switch case in solver
yanchen036 Sep 18, 2015
4c2ff16
caffe.proto: generalize ConvolutionParameter to N spatial axes
jeffdonahue Mar 5, 2015
0813f32
Blob: add SyncedMemory shape accessor for GPU shape access
jeffdonahue Mar 5, 2015
9d8206e
Im2col and Convolution layers support N spatial axes
jeffdonahue Mar 5, 2015
2e1c1cb
Merge pull request #2049 from jeffdonahue/nd-convolution
jeffdonahue Sep 19, 2015
68655b5
Merge pull request #3083 from yanchen036/master
longjon Sep 19, 2015
da75a0e
[build] check xcode command line tools version >= 6
shelhamer Sep 20, 2015
a40c2a0
fix broken DeconvolutionLayer GPU backward caused by typo
longjon Sep 20, 2015
9dca961
Merge pull request #3095 from longjon/fix-deconv-backward
jeffdonahue Sep 20, 2015
6a00eca
fix broken conv/deconv reshaping caused by reading bottom shape in
longjon Sep 20, 2015
58e4ea4
Merge pull request #3093 from shelhamer/check-xcode-clt-version
shelhamer Sep 21, 2015
d425e59
Merge pull request #3096 from longjon/fix-conv-reshape
longjon Sep 21, 2015
74e1745
[test] TestReshape: check small then large
shelhamer Sep 23, 2015
ae77b15
[test] TestReshape: expect instead of check
shelhamer Sep 23, 2015
b8c81bd
[test] TestReshape: check that shapes actually change
shelhamer Sep 23, 2015
ff6b288
Merge pull request #3112 from shelhamer/test-reshape-harder
shelhamer Sep 24, 2015
37dc63c
Merge pull request #2951 from dplarson/examples_readme_link
shelhamer Sep 24, 2015
84e390c
Allow H5T_INTEGER in HDF5 files
lukeyeager Sep 24, 2015
ebc9963
Modify HDF5DataLayerTest to test H5T_INTEGER data
lukeyeager Aug 26, 2015
349ff65
Merge pull request #2978 from lukeyeager/h5t_integer
jeffdonahue Sep 24, 2015
859f938
Fix generate_sample_data.py - bug from #2978
lukeyeager Sep 24, 2015
674b349
Merge pull request #3115 from lukeyeager/fix-generate-sample-data
jeffdonahue Sep 24, 2015
200bd40
Fix parse_log.sh against "prefetch queue empty" messages
ducha-aiki Sep 25, 2015
6c02c8b
Add argmax_param axis
timmeinhardt Sep 15, 2015
c77d5e5
Implement ArgMaxLayer forward_cpu and reshape for axis param
timmeinhardt Sep 15, 2015
9b2d267
Update ArgMaxLayer documentation for axis param
timmeinhardt Sep 15, 2015
a2a5e22
Generalise ArgMaxLayerTest bottom blob shape
timmeinhardt Sep 15, 2015
def3d3c
Implement ArgMaxLayerTest for axis param
timmeinhardt Sep 15, 2015
aa6900a
Merge pull request #3118 from ducha-aiki/patch-1
shelhamer Sep 25, 2015
3b85440
Merge pull request #3032 from ronghanghu/ban-pythonlayer-in-parallel
shelhamer Sep 25, 2015
bd5f154
Add flag on how host memory is allocated
ronghanghu Sep 16, 2015
ff16f6e
Merge pull request #3073 from ronghanghu/consistent-malloc-free
ronghanghu Sep 26, 2015
96ba513
Fix a typo
Sep 29, 2015
942df00
Merge pull request #3128 from hjss06/patch-1
ronghanghu Sep 29, 2015
5cb40b4
Add versioning for v0.14
lukeyeager Sep 30, 2015
9b903b5
test
drnikolaev Sep 30, 2015
c8073dc
Initial cuDNN v3 support
slayton58 Jul 8, 2015
c089a2e
cuDNN v4 compatibility fix
drnikolaev Oct 1, 2015
01e15d0
Merge pull request #3069 from timmeinhardt/argmax
shelhamer Oct 1, 2015
441bddc
Add CNMeM support
NV-slayton Jun 30, 2015
fd0d640
Cmake support for CNMeM
slayton58 Jul 21, 2015
6aaf7b1
Move to centralized cuDNN handle
slayton58 Jul 23, 2015
4a57fff
Turn off cuDNN in TravisCI script
lukeyeager Aug 27, 2015
62582cb
Fix potential CNMEM_NOT_INITIALIZED errors
slayton58 Aug 28, 2015
98cc023
add badge for travis build and license
kashefy Oct 1, 2015
6eae122
Merge pull request #3133 from kashefy/badge
shelhamer Oct 1, 2015
f0d6549
unused variables warning
drnikolaev Oct 1, 2015
848bfda
Fix invalid argument error in tests
NV-slayton Jul 18, 2015
552a84a
Add a caffe.io.write_mean function to the MATLAB interface
Sep 11, 2015
92dc4e6
Merge pull request #3058 from zoharby/master
ronghanghu Oct 5, 2015
64f948a
SilenceLayer Backward bugfix (fixes #3151)
jeffdonahue Oct 5, 2015
8ef3d63
Merge pull request #3152 from jeffdonahue/silence-backward-fix
jeffdonahue Oct 5, 2015
e061546
fixes BVLC/caffe#3163
eerwitt Oct 7, 2015
04c7c36
Merge pull request #3164 from eerwitt/layer-documentation-sample-fixes
ronghanghu Oct 7, 2015
02630e0
CUDNN config improved
borisfom Oct 8, 2015
ffee008
Initial cuDNN v3 support
slayton58 Jul 8, 2015
17b90a2
Fixing memory handler
borisfom Oct 8, 2015
ee9ab75
Fixing memory handler
borisfom Oct 8, 2015
dfd65ed
CNMEM fix
borisfom Oct 9, 2015
cda2e18
Renamed handler class
borisfom Oct 9, 2015
36eaec8
Reverted rename for now
borisfom Oct 9, 2015
636998d
Merge pull request #1 from borisfom/caffe-0.14-cnmem
borisfom Oct 9, 2015
ab8a646
restored missed functions
borisfom Oct 9, 2015
f642fb6
Merge pull request #2 from borisfom/caffe-0.14-cnmem
borisfom Oct 9, 2015
46cc594
restored use_pool check
borisfom Oct 10, 2015
8740cce
Merge pull request #3 from borisfom/caffe-0.14-cnmem
borisfom Oct 10, 2015
d82e665
Merged diffs from NVIDIA/caffe-0.14 and made all tests work.
drnikolaev Oct 12, 2015
ba10864
perparing for merge
borisfom Oct 12, 2015
2b395e9
Merge remote-tracking branch 'slayton/cudnnV3' into caffe-0.14
slayton58 Oct 12, 2015
3931c20
Merged Simon's V3 branch and fixed compiler warnings
drnikolaev Oct 12, 2015
908f082
488f1264805538df1282c143cc5acbf0eb694afc
NV-slayton Jun 30, 2015
f1928b9
438add5c3e55dd6e96f0eb7ffa54cde70f478db6
slayton58 Jul 21, 2015
83b4124
38d6baff443217b91f2b39afa1a1158b3c9172aa
slayton58 Jul 23, 2015
350658d
Fix potential CNMEM_NOT_INITIALIZED errors
slayton58 Aug 28, 2015
5a02f49
cnmem merge
borisfom Oct 12, 2015
6ba96ee
cnmem merge
borisfom Oct 13, 2015
14eaf93
Merged CNMEM, tests passed
borisfom Oct 13, 2015
0b70f2a
CNMEM case checked
borisfom Oct 13, 2015
9405436
peer/self ordering fix
drnikolaev Oct 13, 2015
5dcff36
Polished MemoryHandler
borisfom Oct 13, 2015
d5b7d26
LINT fixes
borisfom Oct 14, 2015
00d6a1b
CPU_ONLY fix
borisfom Oct 15, 2015
b2daa56
macro redefinition fix
borisfom Oct 15, 2015
906002c
Names normalized
borisfom Oct 18, 2015
d12b5cc
deleted submodule
borisfom Oct 19, 2015
6480847
Squashed '.3rdparty/cnmem/' content from commit e817a7a
borisfom Oct 19, 2015
d25e6a5
Merge commit '6480847b61b0f004bfd69616cd82761a64711b61' as '.3rdparty…
borisfom Oct 19, 2015
8a16cbd
CNMEM integration
borisfom Oct 20, 2015
8635051
Travis build has Cmake 2.8.7
borisfom Oct 20, 2015
ce1c463
Removing Cmake
borisfom Oct 20, 2015
3f462d1
CNMEM conditional include fixed
borisfom Oct 20, 2015
80af4b1
Rectified cnmem cmake, renamed 3rd party targets
borisfom Oct 20, 2015
7d5d1cf
Fixing CUDNN options
borisfom Oct 20, 2015
e191830
Cleanup for PR submission
borisfom Oct 20, 2015
0792af7
Removed submodules remainders
borisfom Oct 20, 2015
c3cd5c8
Merge pull request #42 from drnikolaev/caffe-0.14
lukeyeager Oct 21, 2015
ec192dd
Merge pull request #39 from lukeyeager/nvidia/versioning
lukeyeager Oct 21, 2015
f432085
Mark v0.14.0-beta
lukeyeager Oct 21, 2015
c3cb945
Remove the 4D constraint of blobproto IO in python
sh1r0 Oct 8, 2015
a0ac905
Allow old-style shape in blobproto_to_array
lukeyeager Oct 15, 2015
d2c063c
Fixed drawing problems with repeated convolution
DrSleep Oct 14, 2015
61379dc
clean up logging for Net init
shelhamer Oct 17, 2015
b1e2caf
Qualify messages issued by CMake when CUDA is unavailable
eelstork Oct 19, 2015
6f57970
Re-ordering some lines in build files
lukeyeager Sep 25, 2015
749a636
Add ALLOW_LMDB_NOLOCK build option
lukeyeager Sep 25, 2015
4617bd6
Remove versioned symlinks from tools
lukeyeager Oct 22, 2015
8468d18
Merge pull request #44 from lukeyeager/nvidia/cherry-picks
lukeyeager Oct 22, 2015
aaf7ff0
Merge pull request #46 from lukeyeager/nvidia/cmake-too-much-versioning
lukeyeager Oct 22, 2015
013db36
Mark v0.14.0-beta.1
lukeyeager Oct 22, 2015
74a8f12
Added batch normalization layer with test and examples
ducha-aiki Feb 25, 2015
e48dcda
Cleanup batch norm layer, include global stats computation
cdoersch Oct 6, 2015
d52bc27
Fix cuDNN vs. cublas copy+paste error
lukeyeager Oct 29, 2015
285f900
32-bit LMDB fix
borisfom Oct 29, 2015
ae1ac60
Merge pull request #52 from borisfom/caffe-0.14-lmdb-fix
borisfom Oct 30, 2015
9068a9e
Merge pull request #56 from lukeyeager/nvidia/fix-destroy-cudnn
slayton58 Oct 30, 2015
713b02f
Properly set GPU arena in test and time functions
Oct 31, 2015
3cb5b8f
Merge pull request #59 from thatguymike/arena_fix
thatguymike Oct 31, 2015
36385b7
Added CUB memory pool. CNMEM pool kept as default if USE_CNMEM set.
borisfom Oct 28, 2015
cce736f
fixed CPU only
borisfom Nov 1, 2015
fc52f18
Merge pull request #62 from borisfom/caffe-0.14-cub-alloc
borisfom Nov 1, 2015
715ef61
Merge pull request #51 from lukeyeager/nvidia/batchnorm
thatguymike Nov 3, 2015
d7d568a
CuDNNConvolutionLayer accumulate gradients
ronghanghu Nov 3, 2015
d4236c3
Merge pull request #64 from lukeyeager/nvidia/fix-gradient-accumulation
lukeyeager Nov 3, 2015
4486988
Mark v0.14.0-rc.1
lukeyeager Nov 3, 2015
9aef99d
Revert "Mark v0.14.0-rc.1"
lukeyeager Nov 3, 2015
1fafd7f
TravisCI: wget cmake with --no-check-certificate
lukeyeager Nov 3, 2015
ba220d5
Merge pull request #66 from lukeyeager/fix-travis-cmake-download
lukeyeager Nov 3, 2015
c2aae00
Fixing CMake build
borisfom Nov 3, 2015
6233c68
Merge pull request #65 from borisfom/caffe-0.14-workspaces-fix
lukeyeager Nov 4, 2015
560d270
Restored missed CNMEM switches
borisfom Nov 4, 2015
5dcaa09
Added gpu_memory::buffer class to help in no-pool config
borisfom Nov 3, 2015
1c91b46
Merge pull request #69 from borisfom/caffe-0.14-workspaces-fix
lukeyeager Nov 4, 2015
83401d6
Merge pull request #68 from borisfom/caffe-0.14-tempbuf-fix
lukeyeager Nov 4, 2015
6ef51a4
restored 95% of memory grabbed by CNMEM
borisfom Nov 4, 2015
b14de8d
Merge pull request #70 from borisfom/caffe-0.14-cnmem-95
lukeyeager Nov 4, 2015
6804029
Mark v0.14.0-rc.1
lukeyeager Nov 4, 2015
4f02a0b
[travis] fix boost/python3 conda conflict
longjon Nov 5, 2015
62ad22d
Merge pull request #71 from lukeyeager/nvidia/fix-py3-travis-boost
lukeyeager Nov 5, 2015
18fc267
Adding out of memory handler, making arena bins more detailed
borisfom Nov 9, 2015
934b378
Merge pull request #73 from borisfom/caffe-0.14
borisfom Nov 10, 2015
a1fbccd
Change multi-solver handling by dividing user batch by gpu count, whi…
Nov 18, 2015
66d5580
Update computation of variance and global stats in BatchNormLayer
kkhoot Nov 7, 2015
f31002d
Make backward pass work when global stats is active for BatchNormLayer
kkhoot Nov 12, 2015
2eeec26
Merge pull request #78 from thatguymike/divide_batch_by_solvers
lukeyeager Nov 19, 2015
31ee158
Merge pull request #79 from lukeyeager/nvidia/fix-batchnorm
lukeyeager Nov 19, 2015
54811d3
Removing CNMEM switches
borisfom Nov 14, 2015
3bfc708
Merge pull request #82 from borisfom/caffe-0.14
lukeyeager Nov 20, 2015
e0e6432
Fix gradient test to handle batch size correctly now that we divide
Nov 21, 2015
e1680b0
Fix gradient test to handle batch size correctly now that we divide
Nov 21, 2015
35d07ea
Fix travis and test issue
Nov 21, 2015
49f752d
Merge pull request #83 from thatguymike/divide_batch_by_solvers
thatguymike Nov 21, 2015
1664af4
Cleanup docs to reflect multigpu batch change
Nov 23, 2015
1206d56
Merge pull request #84 from thatguymike/multigpudocupdate
lukeyeager Nov 23, 2015
ff38824
Add cudnn v4 batch normalization integration
andrei-pokrovsky-nv Nov 24, 2015
e9f8357
Merge pull request #80 from andrei-pokrovsky/caffe-0.14
thatguymike Nov 24, 2015
72760f2
Mark v0.14.0-rc.2
lukeyeager Nov 24, 2015
155beaf
CUB allocator improvements: no guessing in get_info, infinite pool, l…
borisfom Dec 2, 2015
c742401
Merge remote-tracking branch 'upstream/caffe-0.14' into caffe-0.14
borisfom Dec 2, 2015
4cd523f
Merge pull request #86 from borisfom/caffe-0.14
borisfom Dec 2, 2015
7bd0991
Revert "Pool allocator fixes"
thatguymike Dec 2, 2015
3c378a8
Merge pull request #87 from NVIDIA/revert-86-caffe-0.14
thatguymike Dec 2, 2015
1fdefe9
Dynamically reshape and adjust pool
Dec 2, 2015
4e936f7
Fixed large allocations cache
borisfom Dec 2, 2015
b61ebf0
Fixing GPU arena setup
borisfom Dec 2, 2015
37d8e76
Fixed debug print
borisfom Dec 2, 2015
082bf14
Merge pull request #88 from thatguymike/cub_usage_fixes
thatguymike Dec 2, 2015
64aa3a1
Merge with upstream
borisfom Dec 2, 2015
1c67845
d
borisfom Dec 2, 2015
b52ab92
Cross merge fixes from Boris
thatguymike Dec 3, 2015
b000617
Merge pull request #90 from thatguymike/boris_cub_merge
thatguymike Dec 3, 2015
81baeda
adding environment switch for memory diagnostic
borisfom Dec 3, 2015
db610fd
Mark v0.14.0-rc.3
lukeyeager Dec 3, 2015
d632465
Commencing the merge from upstream
borisfom Dec 7, 2015
1e51bbd
Merge from upstream
borisfom Jan 12, 2016
3fc98dd
Mark v0.14.0-rc.3
lukeyeager Dec 3, 2015
8a2ab8e
gpu_memory cleanup, fixes possible init issue
borisfom Jan 13, 2016
a3d7b22
Merge branch 'memory-plan' into caffe-0.14
borisfom Jan 13, 2016
e1fe549
Fixing Lint errors
borisfom Jan 14, 2016
8eebbb4
Merge pull request #101 from borisfom/caffe-0.14
drnikolaev Jan 15, 2016
569aa49
cuDNN and pooling fixes
drnikolaev Jan 18, 2016
b3a4a1a
Merge pull request #102 from drnikolaev/pooling-cudnn-fixes
borisfom Jan 19, 2016
3724101
Cleaner workaround for max pool
thatguymike Jan 19, 2016
4d58e6d
Merge pull request #103 from thatguymike/maxpool_workaround
drnikolaev Jan 20, 2016
fc6dea9
Fix commentary
thatguymike Jan 20, 2016
c960bcf
Merge pull request #104 from thatguymike/maxpool_commentary
thatguymike Jan 20, 2016
2e4d2ae
Mark v0.14.0
lukeyeager Jan 20, 2016
4440513
weight updates should be done with beta=1
drnikolaev Jan 21, 2016
b9b83d4
Merge pull request #105 from drnikolaev/fix2
thatguymike Jan 21, 2016
c887de9
Mark v0.14.1
lukeyeager Jan 21, 2016
b0074e1
200169109 fix
drnikolaev Jan 29, 2016
3ba0387
Merge pull request #108 from drnikolaev/nvcaffe-0.14-orig
lukeyeager Jan 29, 2016
29361c7
Mark v0.14.2
lukeyeager Jan 29, 2016
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
The table of contents is too big for display.
Diff view
Diff view
  •  
  •  
  •  
2 changes: 2 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,9 @@ Makefile.config
data/*
models/*
*.caffemodel
*.caffemodel.h5
*.solverstate
*.solverstate.h5
*.binaryproto
*leveldb
*lmdb
Expand Down
25 changes: 18 additions & 7 deletions .travis.yml
Original file line number Diff line number Diff line change
Expand Up @@ -2,28 +2,39 @@
# one using CMake, and one using make.
env:
matrix:
- WITH_CUDA=false WITH_CMAKE=false
- WITH_CUDA=false WITH_CMAKE=true
- WITH_CUDA=true WITH_CMAKE=false
- WITH_CUDA=true WITH_CMAKE=true
- WITH_CUDA=false WITH_CMAKE=false WITH_IO=true
- WITH_CUDA=false WITH_CMAKE=true WITH_IO=true PYTHON_VERSION=3
- WITH_CUDA=true WITH_CMAKE=false WITH_IO=true
- WITH_CUDA=true WITH_CMAKE=true WITH_IO=true
- WITH_CUDA=false WITH_CMAKE=false WITH_IO=false
- WITH_CUDA=false WITH_CMAKE=true WITH_IO=false PYTHON_VERSION=3
# Currently there is no way to install cudnn via apt-get. Uncomment when it's available.
# - WITH_CUDA=true WITH_CMAKE=false WITH_IO=true WITH_CUDNN=true
# - WITH_CUDA=true WITH_CMAKE=true WITH_IO=true WITH_CUDNN=true

language: cpp

# Cache Ubuntu apt packages.
cache: apt
cache:
apt: true
directories:
- /home/travis/miniconda
- /home/travis/miniconda2
- /home/travis/miniconda3

compiler: gcc

before_install:
- export NUM_THREADS=4
- export SCRIPTS=./scripts/travis
- export CONDA_DIR="/home/travis/miniconda$PYTHON_VERSION"

install:
- sudo -E $SCRIPTS/travis_install.sh

before_script:
- export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib:/usr/local/cuda/lib64
- export PATH=/home/travis/miniconda/bin:$PATH
- export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib:/usr/local/cuda/lib64:$CONDA_DIR/lib
- export PATH=$CONDA_DIR/bin:$PATH
- if ! $WITH_CMAKE; then $SCRIPTS/travis_setup_makefile_config.sh; fi

script: $SCRIPTS/travis_build_and_test.sh
Expand Down
Loading