OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
821ba6b51b
pytorch/torchgen
History
Driss Guessous b291c1213a Create native function for determining which implementation of SDP to call (#89029) 
# Summary
Creates a callable native function that can determine which implementation of scaled dot product will get called. This allows to bump re-order the runtime dispatch of SDP to enable autograd.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/89029
Approved by: https://github.com/cpuhrsch
2022-11-16 03:07:54 +00:00
..
api Reland 2 Many symintifications (#87604) (#87980) 2022-10-28 13:40:11 +00:00
decompositions
dest Add use_lazy_shape flag to GenLazyIr class (#88444) 2022-11-04 08:23:56 +00:00
operator_versions Apply ufmt linter to all py files under torchgen (#81570) 2022-07-16 03:52:25 +00:00
selective_build [torchgen] Fix selective build error on custom namespace (#83141) 2022-08-10 21:27:05 +00:00
shape_functions [JIT SSA] Allow updating shape functions without recompilation (#83629) 2022-08-22 18:03:44 +00:00
static_runtime [static-runtime] run codegen (#87534) 2022-10-25 23:48:16 +00:00
__init__.py Package config/template files with torchgen (#78942) 2022-06-07 13:33:55 +00:00
BUCK.oss rename BUILD.buck to BUCK.oss (#78792) 2022-06-03 07:23:16 +00:00
BUILD.bazel
build.bzl
code_template.py Apply ufmt linter to all py files under torchgen (#81570) 2022-07-16 03:52:25 +00:00
context.py Change API type Tensor[] for structured kernels. (#73350) 2022-09-26 21:46:38 +00:00
gen_backend_stubs.py Revert "Revert "Symintifying slice ops (#85196)"" (#85746) 2022-09-28 04:37:35 +00:00
gen_functionalization_type.py SymIntify _copy functionalization kernels (and _copy_out too) (#88572) 2022-11-07 21:40:10 +00:00
gen_lazy_tensor.py Add use_lazy_shape flag to GenLazyIr class (#88444) 2022-11-04 08:23:56 +00:00
gen_vmap_plumbing.py Move functorch C++ into aten/src/ATen/functorch (#85381) 2022-09-21 15:50:44 +00:00
gen.py SymIntify _copy functionalization kernels (and _copy_out too) (#88572) 2022-11-07 21:40:10 +00:00
local.py Change API type Tensor[] for structured kernels. (#73350) 2022-09-26 21:46:38 +00:00
model.py PyDispatcher integration with functorch (#88785) 2022-11-16 00:46:59 +00:00
native_function_generation.py Create native function for determining which implementation of SDP to call (#89029) 2022-11-16 03:07:54 +00:00
utils.py autograd: fix non-deterministic output in codegen comments (#84695) 2022-09-13 18:41:15 +00:00