Home
last modified time | relevance | path

Searched full:autogradcuda (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/pytorch/c10/test/core/
H A DDispatchKeySet_test.cpp38 // Similarly, the AutogradCUDA key gets 2 bits in the keyset: in TEST()
40 auto autograd_cuda = DispatchKeySet(DispatchKey::AutogradCUDA); in TEST()
72 {DispatchKey::AutogradCUDA, in TEST()
78 DispatchKeySet({DispatchKey::AutogradCUDA, DispatchKey::CPU})); in TEST()
102 ASSERT_EQ(*autograd_dense_cpu_cuda_iter++, DispatchKey::AutogradCUDA); in TEST()
175 // AutogradCPU, AutogradCUDA
275 // per-backend functionalities. E.g. {AutogradCUDA, CUDA} or in TEST()
276 // {AutogradCPU, AutogradCUDA} There should be 2 total runtime keys in in TEST()
285 // runtime keys to be in the set. e.g. if i = AutogradCUDA, and j = CPU, in TEST()
286 // then combined = {AutogradCUDA, AutogradCPU, CUDA, CPU} in TEST()
[all …]
/aosp_15_r20/external/pytorch/test/
H A Dtest_dispatch.py39 "AutogradCUDA",
424 AutogradCUDA: default_def_name_t_t [math kernel]
467 AutogradCUDA: default_def_name_t_t [math kernel]
509 AutogradCUDA: impl_t_t [math kernel]
556 AutogradCUDA: fn_math [math kernel]
594 AutogradCUDA: impl_t_t [autograd kernel]
645 AutogradCUDA: fn_math [math kernel]
691 AutogradCUDA: fn_math [math kernel]
738 AutogradCUDA: registered in pytorch framework [backend fallback]
787 AutogradCUDA: fn_autograd [autograd kernel]
[all …]
H A Dtest_autograd.py13858 # registrations in derivatives.yaml for Default, AutogradCUDA and NestedTensorAutograd
13867 # bogus gradient registered for AutogradCUDA is grad * 2
13929 # dispatch for _test_autograd_multiple_dispatch.fullcoverage and not AutogradCUDA
13964 # gradient registered to AutogradCUDA is grad.reshape_as(self) + 1
H A Dtest_custom_ops.py1811 @parametrize("key", ["Autograd", "AutogradCPU", "AutogradCUDA"])
/aosp_15_r20/external/pytorch/c10/core/
H A DDispatchKeySet.h84 // - "Autograd": AutogradCPU, AutogradCUDA, Autograd XLA, ...
369 // DispatchKeySet([DispatchKey.CPU, DispatchKey.AutogradCUDA,
370 // DispatchKey.CUDA]).remove(DispatchKey.AutogradCUDA)
372 // DispatchKey.AutogradCUDA]).remove(DispatchKey.AutogradCUDA)
522 // DispatchKey::AutogradCUDA (CPU first because it has lower precedence than
757 constexpr auto autograd_cuda_ks = DispatchKeySet(DispatchKey::AutogradCUDA);
H A DDispatchKey.cpp359 {"AutogradCUDA", c10::DispatchKey::AutogradCUDA}, in parseDispatchKey()
H A DDispatchKeySet.cpp118 case DispatchKey::AutogradCUDA: in getBackendKeySetFromAutograd()
H A DBackend.h71 } else if (t == DispatchKey::CUDA || t == DispatchKey::AutogradCUDA) { in dispatchKeyToBackend()
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DTensorWrapper.cpp58 DispatchKey::AutogradCPU, DispatchKey::AutogradCUDA, DispatchKey::AutogradXLA}); in makeTensorWrapperPtr()
78 DispatchKey::AutogradCPU, DispatchKey::AutogradCUDA, DispatchKey::AutogradXLA}); in unsafeMakeTensorWrapper()
/aosp_15_r20/external/pytorch/torch/_higher_order_ops/
H A Dtriton_kernel_wrap.py741 triton_kernel_wrapper_mutation.fallthrough(DispatchKey.AutogradCUDA)
750 triton_kernel_wrapper_functional.fallthrough(DispatchKey.AutogradCUDA)
751 triton_kernel_wrapper_functional.fallthrough(DispatchKey.AutogradCUDA)
H A Dauto_functionalize.py222 auto_functionalized.fallthrough(DispatchKey.AutogradCUDA)
251 auto_functionalized_v2.fallthrough(DispatchKey.AutogradCUDA)
H A Deffects.py193 with_effects.fallthrough(DispatchKey.AutogradCUDA)
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DVariableFallbackKernel.cpp75 TORCH_LIBRARY_IMPL(_, AutogradCUDA, m) { in TORCH_LIBRARY_IMPL() argument
/aosp_15_r20/external/pytorch/torch/testing/_internal/optests/
H A Dautograd_registration.py92 key = "AutogradCUDA"
/aosp_15_r20/external/pytorch/torch/_custom_op/
H A Dimpl.py326 for key in ["Autograd", "AutogradCPU", "AutogradCUDA"]:
/aosp_15_r20/external/pytorch/torch/export/
H A Dexported_program.py162 torch._C.DispatchKey.AutogradCUDA,
/aosp_15_r20/external/pytorch/torchgen/
H A Dmodel.py211 AutogradCUDA = auto() variable in DispatchKey
322 DispatchKey.AutogradCUDA,
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_dispatch.cpp681 // AutogradCUDA in initDispatchBindings()
/aosp_15_r20/external/pytorch/torch/
H A D_ops.py897 DispatchKey.AutogradCUDA,
/aosp_15_r20/external/pytorch/aten/src/ATen/core/op_registration/
H A Dop_registration_test.cpp1698 // AutogradCUDA is fallthrough, calls CUDA kernel in TEST()
1814 // Similar to in-tree AutogradCPU/AutogradCUDA etc, out-of-tree backends usually register in TEST()
/aosp_15_r20/external/pytorch/torch/_decomp/
H A Ddecompositions.py309 @aten.rrelu_with_noise.default.py_impl(DispatchKey.AutogradCUDA)
333 @aten.rrelu_with_noise_.default.py_impl(DispatchKey.AutogradCUDA)
/aosp_15_r20/external/pytorch/tools/autograd/
H A Dderivatives.yaml2936 AutogradCUDA:
2948 AutogradCUDA:
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_eager_transforms.py5045 @mysum.py_impl(torch._C.DispatchKey.AutogradCUDA)