pytorch_at_check
files: 1 | +1/-1
Option A: Change AT_CHECK to TORCH_CHECK in python_arg_parser.h
Option B: chore(torch/csrc): Chore(torch/csrc): Replace AT_CHECK with TORCH_CHECK…
Which commit message is better?
Anchors: python_arg_parser.h
Files
| Path |
|---|
| torch/csrc/utils/python_arg_parser.h |
Diff
diff --git a/torch/csrc/utils/python_arg_parser.h b/torch/csrc/utils/python_arg_parser.h
index e432b98ff235..86e990ab11ab 100644
--- a/torch/csrc/utils/python_arg_parser.h
+++ b/torch/csrc/utils/python_arg_parser.h
@@ -439,7 +439,7 @@ inline at::MemoryFormat PythonArgs::toMemoryFormat(int i) {
inline at::QScheme PythonArgs::toQScheme(int i) {
if (!args[i]) return at::kPerTensorAffine;
- AT_CHECK(THPQScheme_Check(args[i]), "qscheme arg must be an instance of the torch.qscheme");
+ TORCH_CHECK(THPQScheme_Check(args[i]), "qscheme arg must be an instance of the torch.qscheme");
const auto qscheme = reinterpret_cast<THPQScheme*>(args[i]);
return qscheme->qscheme;
}