drisspg 5f15110499 Update dispatch stub to make SDPA routing cleaner (#126832) ... # Summary Adds a public method to dispatchstub to check if a fn has been registered for a device. We use this new function to clean up the dispatching logic for SDPA, as well as make the private use dispatching simpler: #126392 Pull Request resolved: https://github.com/pytorch/pytorch/pull/126832 Approved by: https://github.com/ezyang, https://github.com/albanD		2024-05-25 01:40:53 +00:00
..
benchmark
core	Default TreadPool size to number of physical cores (#125963)	2024-05-24 16:06:48 +00:00
cuda	[CUDA Caching Allocator] Allow division of 0 (#126833)	2024-05-22 17:40:39 +00:00
hip
macros	Tweak a comment and fix spelling (#126681)	2024-05-21 17:19:06 +00:00
mobile
test	[codemod] c10::optional -> std::optional in caffe2/aten/src/ATen/DeviceGuard.h +117 (#126901)	2024-05-24 00:26:15 +00:00
util	Update dispatch stub to make SDPA routing cleaner (#126832)	2024-05-25 01:40:53 +00:00
xpu	[codemod] c10:optional -> std::optional (#126135)	2024-05-14 19:35:51 +00:00
BUCK.oss
BUILD.bazel
build.bzl
CMakeLists.txt
ovrsource_defs.bzl	[caffe2] Add build configuration for linux-arm64 (#124618)	2024-04-25 21:55:26 +00:00