OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Delete backwards compatibility Backend overload for registerOp (#25914)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/25914

Signed-off-by: Edward Z. Yang <ezyang@fb.com>

Test Plan: Imported from OSS

Differential Revision: D17284083

Pulled By: ezyang

fbshipit-source-id: 430ac7ea2bd042b1f4bb874e53679d0fde326dec
This commit is contained in:
Edward Yang 2019-09-25 07:19:55 -07:00 committed by Facebook Github Bot
parent 3346759774
commit b56ad744a2
2 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
aten/src/ATen/core/ATenDispatch.h
View File

@ -92,11 +92,6 @@ class CAFFE2_API ATenDispatch {
return *this;
}
template<class FuncType>
ATenDispatch& registerOp(Backend b, const char* schema, FuncType* fn) {
return registerOp(backendToTensorTypeId(b), schema, fn);
}
const ATenOpTable* getOpTable(const char* schema) const {
auto iter = op_tables_.find(schema);
TORCH_CHECK(iter != op_tables_.end(),

2
test/cpp_extensions/complex_registration_extension.cpp
View File

@ -47,7 +47,7 @@ static Tensor empty_complex(IntArrayRef size, const TensorOptions & options, c10
}
static auto& complex_empty_registration = globalATenDispatch().registerOp(
Backend::ComplexCPU,
TensorTypeId::ComplexCPUTensorId,
"aten::empty.memory_format(int[] size, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor",
&empty_complex);