frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-05 08:24:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Enable JIT tests on Windows (#27029)

Browse Source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/27029

Reviewed By: eellison

Differential Revision: D20458664

Pulled By: jamesr66a

fbshipit-source-id: 22be918543703869f471e89b3478423198351bf3
This commit is contained in:
peter
2020-03-16 11:18:16 -07:00
committed by Facebook GitHub Bot
parent 1af6002321
commit 24c9e61e79
6 changed files with 20 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/csrc/jit/python/init.cpp
View File

@ -98,7 +98,7 @@ bool loadPythonClasses() {
}
} // anonymous namespace
#if !defined(_WIN32) && !defined(__HIP_PLATFORM_HCC__)
#if !defined(__HIP_PLATFORM_HCC__)
TORCH_API void runJITCPPTests(bool runCuda);
#endif
@ -305,7 +305,7 @@ void initJITBindings(PyObject* module) {
.def(
"_jit_pass_create_autodiff_subgraphs",
[](std::shared_ptr<Graph> graph) { CreateAutodiffSubgraphs(graph); })
#if defined(BUILDING_TESTS) && !defined(_WIN32) && !defined(__HIP_PLATFORM_HCC__)
#if defined(BUILDING_TESTS) && !defined(__HIP_PLATFORM_HCC__)
.def(
"_jit_run_cpp_tests",
[](bool runCuda) {