diff options
| author | Peter Goldsborough <peter@goldsborough.me> | 2018-05-09 14:01:19 -0700 |
|---|---|---|
| committer | Edward Z. Yang <ezyang@mit.edu> | 2018-05-09 17:01:19 -0400 |
| commit | 3023dd25f3586d45097500bb5d9187f2c141d44f (patch) | |
| tree | a1e4868d2586e3624ef8964343b01b7067027d8e /tools | |
| parent | ed111619da10a976e06dfe04ac7dbe3fc3c25ebc (diff) | |
| download | pytorch-3023dd25f3586d45097500bb5d9187f2c141d44f.tar.gz pytorch-3023dd25f3586d45097500bb5d9187f2c141d44f.tar.bz2 pytorch-3023dd25f3586d45097500bb5d9187f2c141d44f.zip | |
Use set_type to implement type conversions in C++ API (#7408)
* Use set_type to implement .cuda() in C++ API
* Change C++ module parameter types in place
* Fix bug where batchnorm state was not moved to CUDA
Diffstat (limited to 'tools')
| -rw-r--r-- | tools/cpp_build/libtorch/CMakeLists.txt | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tools/cpp_build/libtorch/CMakeLists.txt b/tools/cpp_build/libtorch/CMakeLists.txt index e546529aab..ee87443f29 100644 --- a/tools/cpp_build/libtorch/CMakeLists.txt +++ b/tools/cpp_build/libtorch/CMakeLists.txt @@ -310,12 +310,13 @@ if (NOT NO_API) set(TORCH_API_TEST_DIR "${TORCH_SRC_DIR}/../test/cpp/api") add_executable(test_api - ${TORCH_API_TEST_DIR}/main.cpp ${TORCH_API_TEST_DIR}/container.cpp - ${TORCH_API_TEST_DIR}/misc.cpp - ${TORCH_API_TEST_DIR}/rnn.cpp ${TORCH_API_TEST_DIR}/integration.cpp + ${TORCH_API_TEST_DIR}/main.cpp + ${TORCH_API_TEST_DIR}/misc.cpp + ${TORCH_API_TEST_DIR}/module.cpp ${TORCH_API_TEST_DIR}/optim.cpp + ${TORCH_API_TEST_DIR}/rnn.cpp ${TORCH_API_TEST_DIR}/serialization.cpp) target_include_directories(test_api |