mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
4d2f6f1bbe
Summary: Trying to reland https://github.com/pytorch/pytorch/pull/17886 since it broke a build and I reverted it Pull Request resolved: https://github.com/pytorch/pytorch/pull/17924 Differential Revision: D14423842 Pulled By: eellison fbshipit-source-id: f219e786bd07f7da3b7f9e866981199f5ccf6318
61 lines
1.4 KiB
C++
61 lines
1.4 KiB
C++
#include <gtest/gtest.h>
|
|
|
|
#include <test/cpp/jit/test_alias_analysis.h>
|
|
#include <test/cpp/jit/test_class_parser.h>
|
|
#include <test/cpp/jit/test_constant_pooling.h>
|
|
#include <test/cpp/jit/test_irparser.h>
|
|
#include <test/cpp/jit/test_misc.h>
|
|
#include <test/cpp/jit/test_netdef_converter.h>
|
|
|
|
using namespace torch;
|
|
using namespace torch::jit;
|
|
using namespace torch::jit::script;
|
|
|
|
#define JIT_TEST(name) \
|
|
TEST(JitTest, name) { \
|
|
test##name(); \
|
|
}
|
|
|
|
JIT_TEST(ADFormulas)
|
|
JIT_TEST(Attributes)
|
|
JIT_TEST(Blocks)
|
|
JIT_TEST(CodeTemplate)
|
|
JIT_TEST(ControlFlow)
|
|
JIT_TEST(CreateAutodiffSubgraphs)
|
|
JIT_TEST(CustomOperators)
|
|
JIT_TEST(Differentiate)
|
|
JIT_TEST(DifferentiateWithRequiresGrad)
|
|
JIT_TEST(DynamicDAG)
|
|
JIT_TEST(EvalModeForLoadedModule)
|
|
JIT_TEST(FromQualString)
|
|
JIT_TEST(InternedStrings)
|
|
JIT_TEST(IValue)
|
|
JIT_TEST(Proto)
|
|
JIT_TEST(RegisterFusionCachesKernel)
|
|
JIT_TEST(SchemaParser)
|
|
JIT_TEST(TopologicalIndex)
|
|
JIT_TEST(TopologicalMove)
|
|
JIT_TEST(SubgraphUtils)
|
|
JIT_TEST(AliasAnalysis)
|
|
JIT_TEST(WriteTracking)
|
|
JIT_TEST(Wildcards)
|
|
JIT_TEST(MemoryDAG)
|
|
JIT_TEST(IRParser)
|
|
JIT_TEST(ConstantPooling)
|
|
|
|
JIT_TEST(NetDefConverter)
|
|
|
|
JIT_TEST(THNNConv)
|
|
JIT_TEST(ATenNativeBatchNorm)
|
|
JIT_TEST(NoneSchemaMatch)
|
|
JIT_TEST(ClassParser)
|
|
|
|
#define JIT_TEST_CUDA(name) \
|
|
TEST(JitTest, name##_CUDA) { \
|
|
test##name(); \
|
|
}
|
|
|
|
JIT_TEST_CUDA(ArgumentSpec)
|
|
JIT_TEST_CUDA(Fusion)
|
|
JIT_TEST_CUDA(GraphExecutor)
|
|
JIT_TEST_CUDA(Interp)
|