tree: d73ad96013b8e63534919c7ffb80c994b7ae3dec [path history] [tgz]
  1. _nvfuser/
  2. ao/
  3. autograd/
  4. backends/
  5. benchmark_utils/
  6. bottleneck_test/
  7. cpp/
  8. cpp_api_parity/
  9. cpp_extensions/
  10. custom_backend/
  11. custom_operator/
  12. distributed/
  13. distributions/
  14. dynamo/
  15. edge/
  16. error_messages/
  17. expect/
  18. export/
  19. forward_backward_compatibility/
  20. functorch/
  21. fx/
  22. inductor/
  23. jit/
  24. jit_hooks/
  25. lazy/
  26. mobile/
  27. nn/
  28. onnx/
  29. onnx_caffe2/
  30. optim/
  31. package/
  32. profiler/
  33. quantization/
  34. scripts/
  35. test_img/
  36. typing/
  37. _test_bazel.py
  38. allowlist_for_publicAPI.json
  39. conftest.py
  40. create_dummy_torchscript_model.py
  41. delete.py
  42. HowToWriteTestsUsingFileCheck.md
  43. linear.py
  44. load_torchscript_model.py
  45. mkl_verbose.py
  46. mkldnn_verbose.py
  47. run_doctests.sh
  48. run_test.py
  49. simulate_nccl_errors.py
  50. test_ao_sparsity.py
  51. test_autocast.py
  52. test_autograd.py
  53. test_binary_ufuncs.py
  54. test_bundled_images.py
  55. test_bundled_inputs.py
  56. test_comparison_utils.py
  57. test_compile_benchmark_util.py
  58. test_complex.py
  59. test_content_store.py
  60. test_cpp_api_parity.py
  61. test_cpp_extensions_aot.py
  62. test_cpp_extensions_jit.py
  63. test_cpp_extensions_open_device_registration.py
  64. test_cuda.py
  65. test_cuda_expandable_segments.py
  66. test_cuda_multigpu.py
  67. test_cuda_nvml_based_avail.py
  68. test_cuda_primary_ctx.py
  69. test_cuda_sanitizer.py
  70. test_cuda_trace.py
  71. test_custom_op_testing.py
  72. test_dataloader.py
  73. test_datapipe.py
  74. test_decomp.py
  75. test_deploy.py
  76. test_determination.py
  77. test_dispatch.py
  78. test_dlpack.py
  79. test_dynamic_shapes.py
  80. test_expanded_weights.py
  81. test_fake_tensor.py
  82. test_flop_counter.py
  83. test_foreach.py
  84. test_function_schema.py
  85. test_functional_autograd_benchmark.py
  86. test_functional_optim.py
  87. test_functionalization.py
  88. test_functionalization_of_rng_ops.py
  89. test_futures.py
  90. test_fx.py
  91. test_fx_experimental.py
  92. test_fx_passes.py
  93. test_fx_reinplace_pass.py
  94. test_hub.py
  95. test_import_stats.py
  96. test_indexing.py
  97. test_itt.py
  98. test_jit.py
  99. test_jit_autocast.py
  100. test_jit_cuda_fuser.py
  101. test_jit_disabled.py
  102. test_jit_fuser.py
  103. test_jit_fuser_legacy.py
  104. test_jit_fuser_te.py
  105. test_jit_legacy.py
  106. test_jit_llga_fuser.py
  107. test_jit_profiling.py
  108. test_jit_simple.py
  109. test_jit_string.py
  110. test_jiterator.py
  111. test_kernel_launch_checks.py
  112. test_legacy_vmap.py
  113. test_license.py
  114. test_linalg.py
  115. test_logging.py
  116. test_masked.py
  117. test_maskedtensor.py
  118. test_matmul_cuda.py
  119. test_meta.py
  120. test_metal.py
  121. test_mkl_verbose.py
  122. test_mkldnn.py
  123. test_mkldnn_fusion.py
  124. test_mkldnn_verbose.py
  125. test_mobile_optimizer.py
  126. test_model_dump.py
  127. test_module_init.py
  128. test_modules.py
  129. test_monitor.py
  130. test_mps.py
  131. test_multiprocessing.py
  132. test_multiprocessing_spawn.py
  133. test_namedtensor.py
  134. test_namedtuple_return_api.py
  135. test_native_functions.py
  136. test_native_mha.py
  137. test_nestedtensor.py
  138. test_nn.py
  139. test_nnapi.py
  140. test_numba_integration.py
  141. test_numpy_interop.py
  142. test_nvfuser_dynamo.py
  143. test_nvfuser_frontend.py
  144. test_openmp.py
  145. test_ops.py
  146. test_ops_fwd_gradients.py
  147. test_ops_gradients.py
  148. test_ops_jit.py
  149. test_optim.py
  150. test_out_dtype_op.py
  151. test_overrides.py
  152. test_package.py
  153. test_per_overload_api.py
  154. test_prims.py
  155. test_proxy_tensor.py
  156. test_pruning_op.py
  157. test_public_bindings.py
  158. test_python_dispatch.py
  159. test_pytree.py
  160. test_quantization.py
  161. test_reductions.py
  162. test_scatter_gather_ops.py
  163. test_schema_check.py
  164. test_segment_reductions.py
  165. test_serialization.py
  166. test_set_default_mobile_cpu_allocator.py
  167. test_shape_ops.py
  168. test_show_pickle.py
  169. test_sort_and_select.py
  170. test_sparse.py
  171. test_sparse_csr.py
  172. test_sparse_semi_structured.py
  173. test_spectral_ops.py
  174. test_stateless.py
  175. test_static_runtime.py
  176. test_subclass.py
  177. test_sympy_utils.py
  178. test_tensor_creation_ops.py
  179. test_tensorboard.py
  180. test_tensorexpr.py
  181. test_tensorexpr_pybind.py
  182. test_testing.py
  183. test_throughput_benchmark.py
  184. test_torch.py
  185. test_transformers.py
  186. test_type_hints.py
  187. test_type_info.py
  188. test_type_promotion.py
  189. test_typing.py
  190. test_unary_ufuncs.py
  191. test_utils.py
  192. test_view_ops.py
  193. test_vulkan.py
  194. test_weak.py
  195. test_xnnpack_integration.py