b8fc24e092 | ||
---|---|---|
.. | ||
CMakeLists.txt | ||
README.md | ||
gtest_assert_float_eq.h | ||
padded_buffer.cpp | ||
padded_buffer.h | ||
test_approx.cpp | ||
test_aten.cpp | ||
test_base.h | ||
test_boundsinference.cpp | ||
test_conv.cpp | ||
test_cpp_codegen.cpp | ||
test_cuda.cpp | ||
test_dynamic_shapes.cpp | ||
test_expr.cpp | ||
test_external_calls.cpp | ||
test_graph_opt.cpp | ||
test_ir_printer.cpp | ||
test_ir_verifier.cpp | ||
test_kernel.cpp | ||
test_llvm.cpp | ||
test_loopnest.cpp | ||
test_memdependency.cpp | ||
test_memplanning.cpp | ||
test_ops.cpp | ||
test_quantization.cpp | ||
test_reductions.cpp | ||
test_registerizer.cpp | ||
test_simplify.cpp | ||
test_te_fuser_pass.cpp | ||
test_type.cpp | ||
test_type_specializations.cpp | ||
test_utils.h | ||
tutorial.cpp |
README.md
TensorExpr C++ Tests
How to add a new test
First, create a new test file. Test files should have be placed in this
directory, with a name that starts with test_
, like test_foo.cpp
.
Here is an example test file you can copy-paste.
#include <test/cpp/tensorexpr/test_base.h>
// Tests go in torch::jit
namespace torch {
namespace jit {
// 1. Test cases are void() functions.
// 2. They start with the prefix `test`
void testCaseOne() {
// ...
}
void testCaseTwo() {
// ...
}
}
}
Then, register your test in tests.h
:
// Add to TH_FORALL_TESTS_CUDA instead for CUDA-requiring tests
#define TH_FORALL_TESTS(_) \
_(ADFormulas) \
_(Attributes) \
...
_(CaseOne) // note that the `test` prefix is omitted.
_(CaseTwo)
We glob all the test files together in CMakeLists.txt
so that you don't
have to edit it every time you add a test. Unfortunately, this means that in
order to get the build to pick up your new test file, you need to re-run
cmake:
python setup.py build --cmake
How do I run the tests?
The following commands assume you are in PyTorch root.
# (re)build the test binary
ninja build/bin/test_tensorexpr
# run
build/bin/test_tensorexpr --gtest_filter='glob_style_filter*'