frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[OpenReg] Rename cpu_fallback_blacklist to cpu_fallback_blocklist (#161603)

Browse Source 
As the title stated.

Related Infos: https://github.com/pytorch/pytorch/pull/158644#discussion_r2301460839
Pull Request resolved: https://github.com/pytorch/pytorch/pull/161603
Approved by: https://github.com/albanD
This commit is contained in:
FFFrog
2025-08-30 18:10:46 +08:00
committed by PyTorch MergeBot
parent 37da7b777b
commit dad2e50ac5
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/cpp_extensions/open_registration_extension/torch_openreg/csrc/aten/native/Minimal.cpp
View File

@ -164,13 +164,13 @@ at::Tensor view(const at::Tensor& self, c10::SymIntArrayRef size) {
// LITERALINCLUDE START: FALLBACK IMPL
void cpu_fallback(const c10::OperatorHandle& op, torch::jit::Stack* stack) {
static const std::unordered_set<c10::OperatorName> cpu_fallback_blacklist = {
static const std::unordered_set<c10::OperatorName> cpu_fallback_blocklist = {
c10::OperatorName("aten::abs", ""),
c10::OperatorName("aten::abs", "out"),
};
const auto& op_name = op.schema().operator_name();
if (cpu_fallback_blacklist.count(op_name)) {
if (cpu_fallback_blocklist.count(op_name)) {
TORCH_CHECK(
false,
"Operator '",