Name		Name	Last commit message	Last commit date
parent directory ..
CMakeLists.txt		CMakeLists.txt
README.md		README.md
any.cpp		any.cpp
autograd.cpp		autograd.cpp
dataloader.cpp		dataloader.cpp
dispatch.cpp		dispatch.cpp
enum.cpp		enum.cpp
expanding-array.cpp		expanding-array.cpp
fft.cpp		fft.cpp
functional.cpp		functional.cpp
grad_mode.cpp		grad_mode.cpp
imethod.cpp		imethod.cpp
inference_mode.cpp		inference_mode.cpp
init.cpp		init.cpp
init_baseline.h		init_baseline.h
init_baseline.py		init_baseline.py
integration.cpp		integration.cpp
jit.cpp		jit.cpp
memory.cpp		memory.cpp
meta_tensor.cpp		meta_tensor.cpp
misc.cpp		misc.cpp
module.cpp		module.cpp
moduledict.cpp		moduledict.cpp
modulelist.cpp		modulelist.cpp
modules.cpp		modules.cpp
namespace.cpp		namespace.cpp
nested.cpp		nested.cpp
nn_utils.cpp		nn_utils.cpp
operations.cpp		operations.cpp
optim.cpp		optim.cpp
optim_baseline.h		optim_baseline.h
optim_baseline.py		optim_baseline.py
ordered_dict.cpp		ordered_dict.cpp
parallel.cpp		parallel.cpp
parallel_benchmark.cpp		parallel_benchmark.cpp
parameterdict.cpp		parameterdict.cpp
parameterlist.cpp		parameterlist.cpp
rnn.cpp		rnn.cpp
sequential.cpp		sequential.cpp
serialize.cpp		serialize.cpp
special.cpp		special.cpp
static.cpp		static.cpp
support.cpp		support.cpp
support.h		support.h
tensor.cpp		tensor.cpp
tensor_cuda.cpp		tensor_cuda.cpp
tensor_flatten.cpp		tensor_flatten.cpp
tensor_indexing.cpp		tensor_indexing.cpp
tensor_options.cpp		tensor_options.cpp
tensor_options_cuda.cpp		tensor_options_cuda.cpp
torch_include.cpp		torch_include.cpp
transformer.cpp		transformer.cpp

README.md

C++ Frontend Tests

In this folder live the tests for PyTorch's C++ Frontend. They use the GoogleTest test framework.

CUDA Tests

To make a test runnable only on platforms with CUDA, you should suffix your test with _CUDA, e.g.

TEST(MyTestSuite, MyTestCase_CUDA) { }

To make it runnable only on platforms with at least two CUDA machines, suffix it with _MultiCUDA instead of _CUDA, e.g.

TEST(MyTestSuite, MyTestCase_MultiCUDA) { }

There is logic in main.cpp that detects the availability and number of CUDA devices and supplies the appropriate negative filters to GoogleTest.

Integration Tests

Integration tests use the MNIST dataset. You must download it by running the following command from the PyTorch root folder:

$ python tools/download_mnist.py -d test/cpp/api/mnist

The required paths will be referenced as test/cpp/api/mnist/... in the test code, so you must run the integration tests from the PyTorch root folder.

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

api

api

README.md

C++ Frontend Tests

CUDA Tests

Integration Tests

Files

api

Directory actions

More options

Directory actions

More options

Latest commit

History

api

Folders and files

parent directory

README.md

C++ Frontend Tests

CUDA Tests

Integration Tests