Skip to content

Files

Latest commit

kshitij12345pytorchmergebot
kshitij12345
and
pytorchmergebot
[fix] nn c++ : segfault in modulelist and moduledict (pytorch#93074)
Jan 27, 2023
68a9853 · Jan 27, 2023

History

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

api

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
CMakeLists.txt
CMakeLists.txt
Oct 6, 2022
README.md
README.md
Sep 22, 2018
any.cpp
any.cpp
Jun 11, 2022
autograd.cpp
autograd.cpp
Oct 18, 2022
dataloader.cpp
dataloader.cpp
Dec 19, 2022
dispatch.cpp
dispatch.cpp
Jun 11, 2022
enum.cpp
enum.cpp
Jun 11, 2022
expanding-array.cpp
expanding-array.cpp
Dec 9, 2021
fft.cpp
fft.cpp
Jun 11, 2022
functional.cpp
functional.cpp
Sep 13, 2022
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
Aug 12, 2021
integration.cpp
integration.cpp
Dec 9, 2021
jit.cpp
jit.cpp
Jun 11, 2022
memory.cpp
memory.cpp
Jul 23, 2021
meta_tensor.cpp
meta_tensor.cpp
Jun 11, 2022
misc.cpp
misc.cpp
Jun 11, 2022
module.cpp
module.cpp
Jan 26, 2023
moduledict.cpp
moduledict.cpp
Jan 27, 2023
modulelist.cpp
modulelist.cpp
Jan 27, 2023
modules.cpp
modules.cpp
Sep 30, 2022
namespace.cpp
namespace.cpp
Jun 11, 2022
nested.cpp
nested.cpp
Sep 28, 2022
nn_utils.cpp
nn_utils.cpp
Nov 15, 2022
operations.cpp
operations.cpp
Oct 19, 2021
optim.cpp
optim.cpp
Jan 26, 2023
optim_baseline.h
optim_baseline.h
Jun 11, 2022
optim_baseline.py
optim_baseline.py
Aug 12, 2021
ordered_dict.cpp
ordered_dict.cpp
Jul 23, 2021
parallel.cpp
parallel.cpp
Jun 11, 2022
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
Jun 11, 2022
sequential.cpp
sequential.cpp
Jun 11, 2022
serialize.cpp
serialize.cpp
Nov 11, 2022
special.cpp
special.cpp
Jun 11, 2022
static.cpp
static.cpp
Oct 3, 2022
support.cpp
support.cpp
Jul 23, 2021
support.h
support.h
Oct 18, 2022
tensor.cpp
tensor.cpp
Jun 11, 2022
tensor_cuda.cpp
tensor_cuda.cpp
Jun 11, 2022
tensor_flatten.cpp
tensor_flatten.cpp
Jun 11, 2022
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.