commit | 923bcfdd276edab35913a381f24dc6a6ed373c76 | [log] [tgz] |
---|---|---|
author | Dmytro Dzhulgakov <dzhulgakov@fb.com> | Mon Oct 16 20:12:55 2017 -0700 |
committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | Mon Oct 16 20:17:39 2017 -0700 |
tree | b274744dd3494ceb1b788021229f20507cf8d4e9 | |
parent | 4ac8ecb76ef2aea628b30e5e8a5c8dff9b67009d [diff] |
Gate engine=NNPACK with nnp_initialize Summary: Somehow we're observing mysterious test failures for some nnpack-related tests with gcc5 only on Travis: https://travis-ci.org/caffe2/caffe2/jobs/288804879 Marat suggested that maybe the machine doesn't have avx2 support. Right now gating is happening for FB-internal only. I think it makes sense to make gating generic. Calling `nnp_initialize` seems like the right way to do so. It returns failure if the hardware is not supported and is a noop after the first call. Reviewed By: Maratyszcza Differential Revision: D6073808 fbshipit-source-id: e684668628b5c635368351114b6c502d2cc81fe4
Caffe2 is a lightweight, modular, and scalable deep learning framework. Building on the original Caffe, Caffe2 is designed with expression, speed, and modularity in mind.
Caffe2 research award competition request for proposals
Please use Github issues (https://github.com/caffe2/caffe2/issues) to ask questions, report bugs, and request new features.
Please participate in our survey (https://www.surveymonkey.com/r/caffe2). We will send you information about new releases and special developer events/webinars.
Caffe2 is released under the Apache 2.0 license. See the NOTICE file for details.