saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/test
History
Jiakai Liu 3c4d57c18b [pytorch][nnc] update external functions for mobile build (#56850) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/56850

This is part of the changes to enable NNC AOT compilation for mobile.
The generated kernels need to call these external functions thus change the declarations to use C linkage when building the mobile runtime.

Added nnc_aten_addmm external function.

ghstack-source-id: 127877411

Test Plan:
- build & CI;
- tested mobile build with stacked PRs;

Reviewed By: ZolotukhinM

Differential Revision: D27897154

fbshipit-source-id: 61d5499d7781a83bd2657859659fd1b5043d6b04
2021-04-30 19:07:19 -07:00
..
backward_compatibility [CUDA graphs] [BC-breaking] Makes torch.cuda.amp.GradScaler scale updates in-place for better composability with graph capture (#55562) 2021-04-30 13:03:05 -07:00
benchmark_utils Add lint for unqualified noqa (#56272) 2021-04-19 13:16:18 -07:00
bottleneck_test
cpp [pytorch][nnc] update external functions for mobile build (#56850) 2021-04-30 19:07:19 -07:00
cpp_api_parity
cpp_extensions [torch distributed] Implementing all_gather_base (#56315) 2021-04-23 14:16:47 -07:00
custom_backend
custom_operator AutoDispatchBelowAutograd takes no arguments. (#56424) 2021-04-21 14:44:12 -07:00
distributed [8/n] [torch/elastic] Add unit tests for _RendezvousState (#56536) 2021-04-30 13:14:52 -07:00
distributions
error_messages
expect Eliminate global usage of torch.set_default_dtype in sparse test (#56393) 2021-04-27 15:23:14 -07:00
fx Maintain submodule references during subgraph rewriting (#55463) 2021-04-30 16:46:44 -07:00
jit [JIT] Enable conv-add-relu fusion as a part of frozen graph optimization (#56580) 2021-04-30 15:29:38 -07:00
jit_hooks Un-ignore F403 in .flake8 (#55838) 2021-04-13 09:24:07 -07:00
mobile Un-ignore F403 in .flake8 (#55838) 2021-04-13 09:24:07 -07:00
onnx Revert "Revert [ONNX] Redesign inplace conversion" (#56675) 2021-04-22 22:22:16 -07:00
optim
package [package] PackageExporter __import__ logic to not parse dynamic cases (#57283) 2021-04-29 14:21:33 -07:00
quantization per_channel fake quant fp16 and fp64 support (#56894) 2021-04-30 13:52:45 -07:00
scripts
test_img
typing Port NumPy typing testing style to PyTorch (#54234) 2021-04-15 01:25:16 -07:00
HowToWriteTestsUsingFileCheck.md
linear.py
run_test.py Pytorch add agent api tests (#56985) 2021-04-29 06:14:39 -07:00
simulate_nccl_errors.py
test_ao_sparse.py [sparsity] Moving the sparsity python files to OSS (#56617) 2021-04-22 14:07:31 -07:00
test_autograd.py Remove unused forward AD flag (#57058) 2021-04-30 07:32:56 -07:00
test_binary_ufuncs.py [special] Add xlog1py (#55138) 2021-04-30 05:51:13 -07:00
test_bundled_images.py [easy] Rename fb::jpeg_decode_to_NCHW to fb::image_decode_to_NCHW (#55857) 2021-04-15 13:44:13 -07:00
test_bundled_inputs.py Add lint for unqualified type: ignore (#56290) 2021-04-21 08:07:23 -07:00
test_complex.py
test_cpp_api_parity.py
test_cpp_extensions_aot.py Fix ordered_dict.h for CUDA on Windows (#55275) 2021-04-07 23:43:35 -07:00
test_cpp_extensions_jit.py Add minidump collection via breakpad (#55647) 2021-04-16 13:05:01 -07:00
test_cuda.py [CUDA graphs] [BC-breaking] Makes torch.cuda.amp.GradScaler scale updates in-place for better composability with graph capture (#55562) 2021-04-30 13:03:05 -07:00
test_cuda_primary_ctx.py
test_dataloader.py [DataLoader] Add generate_state for NumPy seeding (#56797) 2021-04-27 08:14:02 -07:00
test_datapipe.py Add lint for unqualified type: ignore (#56290) 2021-04-21 08:07:23 -07:00
test_determination.py
test_dispatch.py Add lint for unqualified noqa (#56272) 2021-04-19 13:16:18 -07:00
test_expecttest.py
test_foreach.py
test_function_schema.py Add stricter check for function schemas with varargs (#56509) 2021-04-20 20:04:38 -07:00
test_functional_autograd_benchmark.py
test_futures.py Add lint for unqualified type: ignore (#56290) 2021-04-21 08:07:23 -07:00
test_fx.py Added OpInfo for torch.einsum (#56276) 2021-04-27 07:39:38 -07:00
test_fx_experimental.py [FX] fix normalization issues with lists of tensors (#57004) 2021-04-27 20:02:00 -07:00
test_import_time.py Add a test to measure import torch time (#56041) 2021-04-15 00:53:30 -07:00
test_indexing.py Migrate thrust->cub for index put (#55693) 2021-04-27 12:27:09 -07:00
test_jit.py Support type inference of nn.Module methods using PDT (#57165) 2021-04-29 11:09:37 -07:00
test_jit_cuda_fuser.py Fixing error in jit cuda on ROCm: non-constant-expression cannot be n… (#55243) 2021-04-21 16:35:27 -07:00
test_jit_disabled.py
test_jit_fuser.py Enforce PEP263 for PyTorch python codebase (#55346) 2021-04-06 18:31:38 -07:00
test_jit_fuser_legacy.py Un-ignore F403 in .flake8 (#55838) 2021-04-13 09:24:07 -07:00
test_jit_fuser_te.py [nnc] Enable all fuser tests for cpu (#57332) 2021-04-30 10:11:06 -07:00
test_jit_legacy.py Un-ignore F403 in .flake8 (#55838) 2021-04-13 09:24:07 -07:00
test_jit_profiling.py Un-ignore F403 in .flake8 (#55838) 2021-04-13 09:24:07 -07:00
test_jit_simple.py Un-ignore F403 in .flake8 (#55838) 2021-04-13 09:24:07 -07:00
test_jit_string.py
test_kernel_launch_checks.py
test_license.py
test_linalg.py Fix min() and max() for empty tensors (#52565) 2021-04-30 15:55:10 -07:00
test_logging.py
test_metal.py Remove legacy constructor calls from pytorch codebase. (#54142) 2021-04-11 15:45:17 -07:00
test_mkldnn.py Use mkldnn copy for copy_ when self and src are Mkldnn layout (#54248) 2021-04-08 06:35:39 -07:00
test_mobile_optimizer.py [Pytorch Edge] Remove methods_to_optimize arg (#57045) 2021-04-27 14:54:13 -07:00
test_model_dump.py model_dump tool for model inspection (#56868) 2021-04-28 07:33:10 -07:00
test_module_init.py Support factory kwargs in torch.nn modules (#54508) 2021-04-22 16:16:53 -07:00
test_multiprocessing.py Fix mp serialization for integer nn.Parameter on CUDA (#56529) 2021-04-22 16:21:04 -07:00
test_multiprocessing_spawn.py
test_namedtensor.py
test_namedtuple_return_api.py Added linalg.eig, linalg.eigvals (#52491) 2021-04-06 13:53:26 -07:00
test_native_functions.py
test_nn.py Fix a numerical issue of CUDA channels-last SyncBatchNorm (#57077) 2021-04-29 21:38:52 -07:00
test_nnapi.py Add lint for unqualified noqa (#56272) 2021-04-19 13:16:18 -07:00
test_numba_integration.py
test_numpy_interop.py Add lint for unqualified type: ignore (#56290) 2021-04-21 08:07:23 -07:00
test_op_aliases.py [OpInfo] Add opinfo for transpose and its aliases (#56122) 2021-04-25 21:58:16 -07:00
test_openmp.py
test_ops.py Add supported backward_dtype to OpInfo (#56156) 2021-04-30 08:24:58 -07:00
test_optim.py Remove legacy constructor calls from pytorch codebase. (#54142) 2021-04-11 15:45:17 -07:00
test_overrides.py Update test_overrides for gradcheck (#57155) 2021-04-29 07:43:18 -07:00
test_package.py
test_profiler.py Add gzip format support for chrome tracing (#56554) 2021-04-28 12:40:33 -07:00
test_pruning_op.py
test_public_bindings.py
test_pytree.py
test_quantization.py Add lint for unqualified noqa (#56272) 2021-04-19 13:16:18 -07:00
test_reductions.py Fix min() and max() for empty tensors (#52565) 2021-04-30 15:55:10 -07:00
test_segment_reductions.py [torch][segment_reduce] Add support for initial value (#56923) 2021-04-30 18:01:31 -07:00
test_serialization.py
test_set_default_mobile_cpu_allocator.py
test_shape_ops.py Remove legacy constructor calls from pytorch codebase. (#54142) 2021-04-11 15:45:17 -07:00
test_show_pickle.py
test_sort_and_select.py Port topk from THC to ATen, migrate most of sort as well (#55392) 2021-04-27 14:49:41 -07:00
test_sparse.py Revert D27765618: Initial support for sparse complex tensors constructors for CPU/CUDA 2021-04-27 15:48:51 -07:00
test_sparse_csr.py Modernize test-suite in sparse tensor CSR (#56392) 2021-04-27 15:22:17 -07:00
test_spectral_ops.py Remove distutils (#57040) 2021-04-29 12:10:11 -07:00
test_static_runtime.py
test_tensor_creation_ops.py Revert "Remove sync for randperm on small tensors. (#54113)" (#57299) 2021-04-29 17:52:21 -07:00
test_tensorboard.py Add lint for unqualified noqa (#56272) 2021-04-19 13:16:18 -07:00
test_tensorexpr.py [nnc] Enable all fuser tests for cpu (#57332) 2021-04-30 10:11:06 -07:00
test_tensorexpr_pybind.py
test_testing.py enable support for custom error messages in torch.testing (#55890) 2021-04-24 23:37:44 -07:00
test_throughput_benchmark.py
test_torch.py Update DLPack to 0.4 (#55365) 2021-04-30 10:30:05 -07:00
test_type_hints.py Add lint for unqualified type: ignore (#56290) 2021-04-21 08:07:23 -07:00
test_type_info.py
test_type_promotion.py Revert D27765618: Initial support for sparse complex tensors constructors for CPU/CUDA 2021-04-27 15:48:51 -07:00
test_typing.py Port NumPy typing testing style to PyTorch (#54234) 2021-04-15 01:25:16 -07:00
test_unary_ufuncs.py To fix inconsistency of digamma with SciPy (#56689) 2021-04-27 18:36:11 -07:00
test_utils.py Add lint for unqualified type: ignore (#56290) 2021-04-21 08:07:23 -07:00
test_view_ops.py Adding hsplit,vsplit and dsplit methods (#53536) 2021-04-26 09:39:09 -07:00
test_vmap.py
test_vulkan.py Remove legacy constructor calls from pytorch codebase. (#54142) 2021-04-11 15:45:17 -07:00
test_xnnpack_integration.py Remove legacy constructor calls from pytorch codebase. (#54142) 2021-04-11 15:45:17 -07:00