Update

[ghstack-poisoned]
pytorch · ZhiweiYan-96 · Oct 9, 2024 · Oct 12, 2024 · Oct 15, 2024 · Oct 16, 2024
commit f0d657e1590c4b18f1fd8c2c68afef096ce3b4c2
diff --git a/aten/src/ATen/Context.cpp b/aten/src/ATen/Context.cpp
@@ -122,11 +122,11 @@ void Context::setAllowTF32CuDNN(bool b) {
   allow_tf32_cudnn = b;
 }
 
-bool Context::allowTF32oneDNN() const {
+bool Context::allowTF32OneDNN() const {
   return allow_tf32_onednn;
 }
 
-void Context::setAllowTF32oneDNN(bool b){
+void Context::setAllowTF32OneDNN(bool b){
   allow_tf32_onednn = b;
 }
 

diff --git a/aten/src/ATen/Context.h b/aten/src/ATen/Context.h
@@ -320,8 +320,8 @@ class TORCH_API Context {
   void setFloat32MatmulPrecision(const std::string& s);
   bool allowTF32CuDNN() const;
   void setAllowTF32CuDNN(bool);
-  bool allowTF32oneDNN() const;
-  void setAllowTF32oneDNN(bool);
+  bool allowTF32OneDNN() const;
+  void setAllowTF32OneDNN(bool);
   bool allowTF32CuBLAS() const;
   void setAllowTF32CuBLAS(bool);
   Float32MatmulPrecision float32MatmulPrecision() const;

diff --git a/aten/src/ATen/native/mkldnn/xpu/detail/Utils.cpp b/aten/src/ATen/native/mkldnn/xpu/detail/Utils.cpp
@@ -452,7 +452,7 @@ dnnl::memory::format_tag conv_weight_fmt(
 
 void apply_tf32_if_allowed(dnnl::primitive_attr& pattr) {
   auto& ctx = at::globalContext();
-  bool allow_tf32 = ctx.allowTF32oneDNN();
+  bool allow_tf32 = ctx.allowTF32OneDNN();
   if (allow_tf32) {
     pattr.set_fpmath_mode(dnnl::fpmath_mode::tf32);
   }

diff --git a/torch/_C/__init__.pyi.in b/torch/_C/__init__.pyi.in
@@ -1172,8 +1172,8 @@ def _get_cudnn_deterministic() -> _bool: ...  # THPModule_deterministicCuDNN
 def _set_cudnn_deterministic(arg: _bool) -> None: ...  # THPModule_setDeterministicCuDNN
 def _get_mkldnn_deterministic() -> _bool: ...  # THPModule_deterministicMkldnn
 def _set_mkldnn_deterministic(arg: _bool) -> None: ...  # THPModule_setDeterministicMkldnn
-def _get_onednn_allow_tf32() -> _bool: ... # THPModule_allowTF32oneDNN
-def _set_onednn_allow_tf32(arg: _bool) -> None: ... # THPModule_setAllowTF32oneDNN
+def _get_onednn_allow_tf32() -> _bool: ... # THPModule_allowTF32OneDNN
+def _set_onednn_allow_tf32(arg: _bool) -> None: ... # THPModule_setAllowTF32OneDNN
 def _get_deterministic_algorithms() -> _bool: ...  # THPModule_deterministicAlgorithms
 def _get_deterministic_algorithms_warn_only() -> _bool: ...  # THPModule_deterministicAlgorithmsWarnOnly
 def _set_deterministic_algorithms(

diff --git a/torch/csrc/Module.cpp b/torch/csrc/Module.cpp
@@ -921,7 +921,7 @@ static PyObject* THPModule_setDeterministicAlgorithms(
   END_HANDLE_TH_ERRORS
 }
 
-static PyObject* THPModule_setAllowTF32oneDNN(
+static PyObject* THPModule_setAllowTF32OneDNN(
     PyObject* _unsued,
     PyObject* arg) {
   HANDLE_TH_ERRORS
@@ -930,15 +930,15 @@ static PyObject* THPModule_setAllowTF32oneDNN(
       "_set_onednn_allow_tf32 expects a bool, "
       "but got ",
       THPUtils_typename(arg));
-  at::globalContext().setAllowTF32oneDNN(arg == Py_True);
+  at::globalContext().setAllowTF32OneDNN(arg == Py_True);
   Py_RETURN_NONE;
   END_HANDLE_TH_ERRORS
 }
 
-static PyObject* THPModule_allowTF32oneDNN(
+static PyObject* THPModule_allowTF32OneDNN(
     PyObject* _unused,
     PyObject* noargs) {
-  if (at::globalContext().allowTF32oneDNN())
+  if (at::globalContext().allowTF32OneDNN())
     Py_RETURN_TRUE;
   else
     Py_RETURN_FALSE;
@@ -1488,8 +1488,8 @@ static std::initializer_list<PyMethodDef> TorchMethods = {
     {"_set_mkldnn_enabled", THPModule_setUserEnabledMkldnn, METH_O, nullptr},
     {"_get_cudnn_allow_tf32", THPModule_allowTF32CuDNN, METH_NOARGS, nullptr},
     {"_set_cudnn_allow_tf32", THPModule_setAllowTF32CuDNN, METH_O, nullptr},
-    {"_get_onednn_allow_tf32", THPModule_allowTF32oneDNN, METH_NOARGS, nullptr},
-    {"_set_onednn_allow_tf32", THPModule_setAllowTF32oneDNN, METH_O, nullptr},
+    {"_get_onednn_allow_tf32", THPModule_allowTF32OneDNN, METH_NOARGS, nullptr},
+    {"_set_onednn_allow_tf32", THPModule_setAllowTF32OneDNN, METH_O, nullptr},
     {"_get_cudnn_benchmark", THPModule_benchmarkCuDNN, METH_NOARGS, nullptr},
     {"_set_cudnn_benchmark", THPModule_setBenchmarkCuDNN, METH_O, nullptr},
     {"_get_cudnn_deterministic",