mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
71d142604f
Summary: This commit adds the hooks in schema parser for futures, options, mutable alias sets, marking writes, and named output arguments that need to exist for other upcoming work. This also fixes that problem where you could not declare Lists of Lists. Implementation of most of these features is left NYI. This commit should avoid merge conflicts for these individual features on the schema parser. Pull Request resolved: https://github.com/pytorch/pytorch/pull/12585 Differential Revision: D10382229 Pulled By: zdevito fbshipit-source-id: 41d794e58ca462cf3a389861c533c68944dc560b
35 lines
719 B
C++
35 lines
719 B
C++
#include <gtest/gtest.h>
|
|
|
|
#include <test/cpp/jit/tests.h>
|
|
|
|
using namespace torch;
|
|
using namespace torch::jit;
|
|
|
|
#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(SchemaParser)
|
|
JIT_TEST(Differentiate)
|
|
JIT_TEST(DifferentiateWithRequiresGrad)
|
|
JIT_TEST(FromQualString)
|
|
JIT_TEST(InternedStrings)
|
|
JIT_TEST(IValue)
|
|
|
|
#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)
|