Skip to content

Files

Latest commit

mikey dagitsespytorchmergebot
mikey dagitses
and
pytorchmergebot
make TensorImpl::data_ptr_impl() non-const and have mutable in the na…
Apr 9, 2023
9d36361 · Apr 9, 2023

History

History
This branch is 28641 commits behind pytorch/pytorch:main.

jit

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
upgrader_models
upgrader_models
Jan 26, 2023
CMakeLists.txt
CMakeLists.txt
Jan 26, 2023
README.md
README.md
Sep 25, 2020
__init__.py
__init__.py
Jul 18, 2020
script_module_v4.ptl
script_module_v4.ptl
Apr 28, 2021
script_module_v5.ptl
script_module_v5.ptl
Apr 28, 2021
script_module_v6.ptl
script_module_v6.ptl
May 24, 2021
source_range_test.cpp
source_range_test.cpp
Apr 18, 2022
test_add_if_then_else.cpp
test_add_if_then_else.cpp
Feb 17, 2022
test_alias_analysis.cpp
test_alias_analysis.cpp
Jul 27, 2022
test_argument_spec.cpp
test_argument_spec.cpp
Oct 27, 2021
test_autodiff.cpp
test_autodiff.cpp
Mar 29, 2022
test_backend.cpp
test_backend.cpp
Apr 18, 2022
test_backend_compiler_lib.cpp
test_backend_compiler_lib.cpp
Apr 9, 2023
test_backend_compiler_preprocess.cpp
test_backend_compiler_preprocess.cpp
Oct 27, 2021
test_backend_lib.cpp
test_backend_lib.cpp
Jul 23, 2021
test_class_import.cpp
test_class_import.cpp
Oct 19, 2021
test_class_parser.cpp
test_class_parser.cpp
Oct 19, 2021
test_class_type.cpp
test_class_type.cpp
Jul 23, 2021
test_cleanup_passes.cpp
test_cleanup_passes.cpp
Jul 23, 2021
test_code_template.cpp
test_code_template.cpp
Jan 14, 2022
test_concat_opt.cpp
test_concat_opt.cpp
Nov 15, 2021
test_constant_pooling.cpp
test_constant_pooling.cpp
Jul 23, 2021
test_create_autodiff_subgraphs.cpp
test_create_autodiff_subgraphs.cpp
Jul 23, 2021
test_cs_debug_info_serialization.cpp
test_cs_debug_info_serialization.cpp
May 25, 2021
test_custom_class.cpp
test_custom_class.cpp
Aug 29, 2022
test_custom_class_registrations.cpp
test_custom_class_registrations.cpp
Nov 17, 2022
test_custom_class_registrations.h
test_custom_class_registrations.h
Aug 29, 2022
test_custom_operators.cpp
test_custom_operators.cpp
Oct 8, 2021
test_dce.cpp
test_dce.cpp
Jul 23, 2021
test_exception.cpp
test_exception.cpp
Feb 12, 2023
test_file_format.cpp
test_file_format.cpp
Mar 4, 2022
test_flatbuffer.cpp
test_flatbuffer.cpp
Nov 1, 2022
test_fuser.cpp
test_fuser.cpp
Jun 11, 2022
test_graph_executor.cpp
test_graph_executor.cpp
Dec 1, 2022
test_graph_iterator.cpp
test_graph_iterator.cpp
Apr 4, 2022
test_inliner.cpp
test_inliner.cpp
Oct 27, 2021
test_interface.cpp
test_interface.cpp
Oct 19, 2021
test_interpreter.cpp
test_interpreter.cpp
Sep 21, 2021
test_interpreter_async.pt
test_interpreter_async.pt
Nov 5, 2020
test_ir.cpp
test_ir.cpp
Jul 23, 2021
test_irparser.cpp
test_irparser.cpp
Jun 11, 2022
test_jit_logging_levels.cpp
test_jit_logging_levels.cpp
Nov 23, 2022
test_jit_type.cpp
test_jit_type.cpp
Oct 8, 2021
test_lite_interpreter.cpp
test_lite_interpreter.cpp
Mar 27, 2023
test_lite_interpreter_direct.cpp
test_lite_interpreter_direct.cpp
Jan 20, 2022
test_lite_trainer.cpp
test_lite_trainer.cpp
Dec 8, 2022
test_load_upgraders.cpp
test_load_upgraders.cpp
Aug 9, 2022
test_memory_dag.cpp
test_memory_dag.cpp
Jul 23, 2021
test_misc.cpp
test_misc.cpp
Jan 21, 2023
test_mobile_type_parser.cpp
test_mobile_type_parser.cpp
Jan 11, 2022
test_module_api.cpp
test_module_api.cpp
Dec 4, 2022
test_op_replacement.cpp
test_op_replacement.cpp
Dec 25, 2021
test_peephole_optimize.cpp
test_peephole_optimize.cpp
Jul 23, 2021
test_qualified_name.cpp
test_qualified_name.cpp
Jul 23, 2021
test_save_load.cpp
test_save_load.cpp
Dec 29, 2022
test_schema_info.cpp
test_schema_info.cpp
Jul 27, 2022
test_schema_matching.cpp
test_schema_matching.cpp
Aug 30, 2021
test_script_profile.cpp
test_script_profile.cpp
May 25, 2021
test_shape_analysis.cpp
test_shape_analysis.cpp
May 20, 2022
test_stack_opt.cpp
test_stack_opt.cpp
Aug 24, 2021
test_subgraph_matcher.cpp
test_subgraph_matcher.cpp
Jul 23, 2021
test_subgraph_rewriter.cpp
test_subgraph_rewriter.cpp
Sep 24, 2021
test_subgraph_utils.cpp
test_subgraph_utils.cpp
Jul 23, 2021
test_union.cpp
test_union.cpp
Oct 8, 2021
test_upgrader_utils.cpp
test_upgrader_utils.cpp
Jan 12, 2022
test_utils.cpp
test_utils.cpp
Aug 30, 2021
test_utils.h
test_utils.h
Apr 18, 2022
tests_setup.py
tests_setup.py
Feb 12, 2023
torch_python_test.cpp
torch_python_test.cpp
Jun 10, 2022

README.md

JIT C++ Tests

Adding 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.

In general a single test suite

Add your test file to the JIT_TEST_SRCS list in test/cpp/jit/CMakeLists.txt.

A test file may look like:

#include <gtest/gtest.h>

using namespace ::torch::jit

TEST(FooTest, BarBaz) {
   // ...
}

// Append '_CUDA' to the test case name will automatically filter it out if CUDA
// is not compiled.
TEST(FooTest, NeedsAGpu_CUDA) {
   // ...
}

// Similarly, if only one GPU is detected, tests with `_MultiCUDA` at the end
// will not be run.
TEST(FooTest, NeedsMultipleGpus_MultiCUDA) {
   // ...
}

Building and running the tests

The following commands assume you are in PyTorch root.

# ... Build PyTorch from source, e.g.
python setup.py develop
# (re)build just the binary
ninja -C build bin/test_jit
# run tests
build/bin/test_jit --gtest_filter='glob_style_filter*'