#include #include namespace torch { namespace jit { #if defined(_WIN32) void runJITCPPTests(bool runCuda) { TORCH_INTERNAL_ASSERT(false, "JIT tests not yet supported on Windows"); } #else #define JIT_TEST(name) test##name(); TORCH_API void runJITCPPTests(bool runCuda) { TH_FORALL_TESTS(JIT_TEST) if (runCuda) { TH_FORALL_TESTS_CUDA(JIT_TEST) } // This test is special since it requires prior setup in python. // So it is not part of the general test list (which is shared between the gtest // and python test runners), but is instead invoked manually by the // torch_python_test.cpp testEvalModeForLoadedModule(); } #undef JIT_TEST #endif } // namespace jit } // namespace torch