Skip to content

Commit 61def1a

Browse files
spcypptmeta-codesync[bot]
authored andcommitted
Change to warn_once for frontend/backend mismatch (#5097)
Summary: Pull Request resolved: #5097 X-link: https://github.com/facebookresearch/FBGEMM/pull/2104 Use TORCH_WARN_ONCE instead of TORCH_WARN to reduce the warning message. Reviewed By: q10 Differential Revision: D86381414 fbshipit-source-id: 90de545c83e50c6a449ff0852081673c50c80bd8
1 parent 396ad52 commit 61def1a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

fbgemm_gpu/codegen/training/pt2/embedding_split_host_pt2_autograd_template.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1236,7 +1236,7 @@ Tensor {{ bwd_mdesc }}_embedding_codegen_lookup_{{ optimizer }}_function_pt2(
12361236

12371237
{%- if has_gpu_support or has_cpu_support %}
12381238

1239-
TORCH_WARN(aux_tensor.size() <= AUX_TENSOR_SIZE,
1239+
TORCH_WARN_ONCE(aux_tensor.size() <= AUX_TENSOR_SIZE,
12401240
"aux_tensor.size() should not be larger than ",
12411241
AUX_TENSOR_SIZE,
12421242
"but found to be ",

0 commit comments

Comments
 (0)