Skip to content

Files

Latest commit

malfetpytorchmergebot
malfet
and
pytorchmergebot
Enable CI on SM89 (pytorch#140305)
Dec 3, 2024
38bbe37 · Dec 3, 2024

History

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

api

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
CMakeLists.txt
CMakeLists.txt
Oct 25, 2024
README.md
README.md
Sep 22, 2018
any.cpp
any.cpp
Jun 11, 2022
autograd.cpp
autograd.cpp
Jul 15, 2024
dataloader.cpp
dataloader.cpp
Oct 28, 2024
dispatch.cpp
dispatch.cpp
Oct 4, 2024
enum.cpp
enum.cpp
Sep 24, 2023
expanding-array.cpp
expanding-array.cpp
Dec 9, 2021
fft.cpp
fft.cpp
Jun 11, 2022
functional.cpp
functional.cpp
Oct 28, 2024
grad_mode.cpp
grad_mode.cpp
Jun 11, 2022
inference_mode.cpp
inference_mode.cpp
Oct 18, 2022
init.cpp
init.cpp
Jun 11, 2022
init_baseline.h
init_baseline.h
Jun 11, 2022
init_baseline.py
init_baseline.py
Jul 31, 2024
integration.cpp
integration.cpp
Oct 19, 2024
ivalue.cpp
ivalue.cpp
May 17, 2024
jit.cpp
jit.cpp
Jun 11, 2022
memory.cpp
memory.cpp
May 14, 2024
meta_tensor.cpp
meta_tensor.cpp
Jun 11, 2022
misc.cpp
misc.cpp
Jun 11, 2022
module.cpp
module.cpp
Oct 28, 2024
moduledict.cpp
moduledict.cpp
Jan 27, 2023
modulelist.cpp
modulelist.cpp
Jan 27, 2023
modules.cpp
modules.cpp
Oct 19, 2024
namespace.cpp
namespace.cpp
Jun 11, 2022
nested.cpp
nested.cpp
Sep 28, 2022
nested_int.cpp
nested_int.cpp
Feb 16, 2024
nn_utils.cpp
nn_utils.cpp
Oct 19, 2024
operations.cpp
operations.cpp
Oct 19, 2024
optim.cpp
optim.cpp
Oct 19, 2024
optim_baseline.h
optim_baseline.h
Jun 11, 2022
optim_baseline.py
optim_baseline.py
Apr 5, 2024
ordered_dict.cpp
ordered_dict.cpp
Jul 23, 2021
parallel.cpp
parallel.cpp
Oct 28, 2024
parallel_benchmark.cpp
parallel_benchmark.cpp
May 8, 2021
parameterdict.cpp
parameterdict.cpp
Jun 11, 2022
parameterlist.cpp
parameterlist.cpp
Oct 19, 2021
rnn.cpp
rnn.cpp
Dec 3, 2024
sequential.cpp
sequential.cpp
Jun 11, 2022
serialize.cpp
serialize.cpp
May 31, 2024
special.cpp
special.cpp
Jun 11, 2022
static.cpp
static.cpp
Oct 26, 2024
support.cpp
support.cpp
Jul 23, 2021
support.h
support.h
Aug 15, 2023
tensor.cpp
tensor.cpp
Apr 4, 2023
tensor_cuda.cpp
tensor_cuda.cpp
Jun 11, 2022
tensor_flatten.cpp
tensor_flatten.cpp
Oct 16, 2023
tensor_indexing.cpp
tensor_indexing.cpp
Jun 11, 2022
tensor_options.cpp
tensor_options.cpp
Jun 11, 2022
tensor_options_cuda.cpp
tensor_options_cuda.cpp
Jun 11, 2022
torch_include.cpp
torch_include.cpp
Jul 23, 2021
transformer.cpp
transformer.cpp
Jun 11, 2022

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.