mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 13:44:15 +08:00
Revert "add amp support for custom backend (#96188)"
This reverts commit cf12edee02a44009c4f06e36efa97d9a7372ab35. Reverted https://github.com/pytorch/pytorch/pull/96188 on behalf of https://github.com/kit1980 due to Broke some linalg tests : https://github.com/pytorch/pytorch/actions/runs/4420037607/jobs/7750708339
This commit is contained in:
@ -464,7 +464,6 @@ PyObject* THModule_getCppBacktrace(PyObject* _unused, PyObject* args) {
|
||||
c10::get_backtrace(frames_to_skip, maximum_number_of_frames, true));
|
||||
END_HANDLE_TH_ERRORS
|
||||
}
|
||||
|
||||
static PyObject* THModule_rename_privateuse1_backend(
|
||||
PyObject* _unused,
|
||||
PyObject* arg) {
|
||||
@ -480,14 +479,6 @@ static PyObject* THModule_rename_privateuse1_backend(
|
||||
END_HANDLE_TH_ERRORS
|
||||
}
|
||||
|
||||
static PyObject* THModule_get_privateuse1_backend_name(
|
||||
PyObject* _unused,
|
||||
PyObject* arg) {
|
||||
HANDLE_TH_ERRORS
|
||||
return THPUtils_packString(c10::get_privateuse1_backend());
|
||||
END_HANDLE_TH_ERRORS
|
||||
}
|
||||
|
||||
PyObject* THPModule_setAllowTF32CuDNN(PyObject* _unused, PyObject* arg) {
|
||||
THPUtils_assert(
|
||||
PyBool_Check(arg),
|
||||
@ -1143,10 +1134,6 @@ static PyMethodDef TorchMethods[] = {
|
||||
THModule_rename_privateuse1_backend,
|
||||
METH_O,
|
||||
nullptr},
|
||||
{"_get_privateuse1_backend_name",
|
||||
THModule_get_privateuse1_backend_name,
|
||||
METH_NOARGS,
|
||||
nullptr},
|
||||
{"set_flush_denormal", THPModule_setFlushDenormal, METH_O, nullptr},
|
||||
{"get_default_dtype", THPModule_getDefaultDtype, METH_NOARGS, nullptr},
|
||||
{"_get_default_device", THPModule_getDefaultDevice, METH_NOARGS, nullptr},
|
||||
|
Reference in New Issue
Block a user