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 37 Packages Projects Releases Wiki Activity

[nvfuser] prevent spamming warning message (#77777)

Browse Source 
updating TORCH_WARN to TORCH_WARN_ONCE to prevent spamming the log
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77777
Approved by: https://github.com/davidberard98
This commit is contained in:
jjsjann123 2022-05-19 20:43:14 +00:00 committed by PyTorch MergeBot
parent 5e0589ca20
commit 17fbb85734
2 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
torch/csrc/jit/codegen/cuda/graph_fuser.cpp
View File

@ -100,8 +100,8 @@ Value* createConditionalConstant(Node* profile_ivalue) {
// ival
val = IValue(profile_ivalue->ival(Symbol::attr("profiled_ival")));
} else {
GRAPH_DEBUG("profile_ivalue: ", *profile_ivalue);
TORCH_WARN(
GRAPH_DEBUG("no profile info in profile_ivalue node: ", *profile_ivalue);
TORCH_WARN_ONCE(
__func__,
" profile_node ",
*profile_ivalue,

16
torch/csrc/jit/codegen/cuda/parser.cpp
View File

@ -3468,7 +3468,7 @@ void profileReductionSize(ProfilingRecord* pr, Node* node, size_t offset) {
if (profiled_ints.size() != size_vec.size() ||
!std::equal(
profiled_ints.begin(), profiled_ints.end(), size_vec.begin())) {
TORCH_WARN(
TORCH_WARN_ONCE(
__FUNCTION__,
" sees varying value in profiling, ignoring and this should be handled by GUARD logic");
pn->s_(profileFailedAttr, "varying profile values");
@ -3510,7 +3510,7 @@ void profileViewSize(ProfilingRecord* pr, Node* node, size_t offset) {
profiled_ints.begin(),
profiled_ints.end(),
input_ints.begin())) {
TORCH_WARN(
TORCH_WARN_ONCE(
__FUNCTION__,
" sees varying value in profiling, ignoring and this should be handled by GUARD logic");
pn->s_(profileFailedAttr, "varying profile values");
@ -3553,7 +3553,7 @@ void profileIntList(ProfilingRecord* pr, Node* node, size_t offset) {
profiled_ints.begin(),
profiled_ints.end(),
input_ints.begin())) {
TORCH_WARN(
TORCH_WARN_ONCE(
__FUNCTION__,
" sees varying value in profiling, ignoring and this should be handled by GUARD logic");
pn->s_(profileFailedAttr, "varying profile values");
@ -3592,7 +3592,7 @@ void profileString(ProfilingRecord* pr, Node* node, size_t offset) {
const auto& profiled_str = pn->s(strAttr);
const auto& input_str = value.toStringRef();
if (input_str != profiled_str) {
TORCH_WARN(
TORCH_WARN_ONCE(
__FUNCTION__,
" sees varying value in profiling, ignoring and this should be handled by GUARD logic");
pn->s_(profileFailedAttr, "varying profile values");
@ -3631,7 +3631,7 @@ void profileBool(ProfilingRecord* pr, Node* node, size_t offset) {
auto profiled_bool = pn->i(boolAttr);
auto input_bool = value.toBool();
if (input_bool != profiled_bool) {
TORCH_WARN(
TORCH_WARN_ONCE(
__FUNCTION__,
" sees varying value in profiling, ignoring and this should be handled by GUARD logic");
pn->s_(profileFailedAttr, "varying profile values");
@ -3670,7 +3670,7 @@ void profileInt(ProfilingRecord* pr, Node* node, size_t offset) {
auto profiled_int = pn->i(intAttr);
auto input_int = value.toInt();
if (input_int != profiled_int) {
TORCH_WARN(
TORCH_WARN_ONCE(
__FUNCTION__,
" sees varying value in profiling, ignoring and this should be handled by GUARD logic");
pn->s_(profileFailedAttr, "varying profile values");
@ -3707,7 +3707,7 @@ void profileIval(ProfilingRecord* pr, Node* node, size_t offset) {
} else {
auto profiled_ival = pn->ival(ivalAttr);
if (value != profiled_ival) {
TORCH_WARN(
TORCH_WARN_ONCE(
__FUNCTION__,
" sees varying value in profiling, ignoring and this should be handled by GUARD logic");
pn->s_(profileFailedAttr, "varying profile values");
@ -3752,7 +3752,7 @@ void profileBoolList(ProfilingRecord* pr, Node* node, size_t offset) {
input_bools.begin(),
input_bools.end(),
profiled_ints.begin())) {
TORCH_WARN(
TORCH_WARN_ONCE(
__FUNCTION__,
" sees varying value in profiling, ignoring and this should be handled by GUARD logic");
pn->s_(profileFailedAttr, "varying profile values");