torch_python_test.cpp

#include <c10/util/Exception.h>
#include <test/cpp/jit/tests.h>
#include <test/cpp/tensorexpr/tests.h>

namespace torch {
namespace jit {

#ifdef _MSC_VER
#define JIT_TEST_API
#else
#define JIT_TEST_API TORCH_API
#endif

#define JIT_TEST(name) test##name();
JIT_TEST_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();
  testTorchSaveError();
}
#undef JIT_TEST

#define JIT_TEST(name) test##name();
JIT_TEST_API void runTENSOREXPRCPPTests(bool runCuda) {
  TH_FORALL_TENSOREXPR_TESTS(JIT_TEST)
  if (runCuda) {
#ifdef USE_CUDA
    TH_FORALL_TENSOREXPR_TESTS_CUDA(JIT_TEST)
#endif
  }
}
#undef JIT_TEST

} // namespace jit
} // namespace torch