Name		Name	Last commit message	Last commit date
parent directory ..
CMakeLists.txt		CMakeLists.txt
README.md		README.md
gtest_assert_float_eq.h		gtest_assert_float_eq.h
padded_buffer.cpp		padded_buffer.cpp
padded_buffer.h		padded_buffer.h
test_approx.cpp		test_approx.cpp
test_aten.cpp		test_aten.cpp
test_base.h		test_base.h
test_boundsinference.cpp		test_boundsinference.cpp
test_conv.cpp		test_conv.cpp
test_cpp_codegen.cpp		test_cpp_codegen.cpp
test_cuda.cpp		test_cuda.cpp
test_dynamic_shapes.cpp		test_dynamic_shapes.cpp
test_expr.cpp		test_expr.cpp
test_external_calls.cpp		test_external_calls.cpp
test_graph_opt.cpp		test_graph_opt.cpp
test_ir_printer.cpp		test_ir_printer.cpp
test_ir_verifier.cpp		test_ir_verifier.cpp
test_kernel.cpp		test_kernel.cpp
test_llvm.cpp		test_llvm.cpp
test_loopnest.cpp		test_loopnest.cpp
test_memdependency.cpp		test_memdependency.cpp
test_memplanning.cpp		test_memplanning.cpp
test_ops.cpp		test_ops.cpp
test_quantization.cpp		test_quantization.cpp
test_reductions.cpp		test_reductions.cpp
test_registerizer.cpp		test_registerizer.cpp
test_simplify.cpp		test_simplify.cpp
test_te_fuser_pass.cpp		test_te_fuser_pass.cpp
test_type.cpp		test_type.cpp
test_type_specializations.cpp		test_type_specializations.cpp
test_utils.h		test_utils.h
tutorial.cpp		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*'

Files

tensorexpr

Directory actions

More options

Directory actions

More options

Latest commit

History

tensorexpr

Folders and files

parent directory

TensorExpr C++ Tests

How to add a new test

How do I run the tests?