Skip to content

Commit 2a48a40

Browse files
nipung90facebook-github-bot
authored andcommitted
Add the torchrec scuba logger extension of the base scuba logger (pytorch#3142)
Summary: Pull Request resolved: pytorch#3142 This is the wrapper around the base logger to create the torchrec specific instance of the logger. This is where the scuba table name used for torchrec logging can be specified. Reviewed By: kausv, saumishr Differential Revision: D76294253 fbshipit-source-id: 214f241269dd089253498c7f7cbc5122299a43a0
1 parent 643d221 commit 2a48a40

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed
Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
#!/usr/bin/env python3
2+
# Copyright (c) Meta Platforms, Inc. and affiliates.
3+
# All rights reserved.
4+
#
5+
# This source code is licensed under the BSD-style license found in the
6+
# LICENSE file in the root directory of this source tree.
7+
8+
9+
import logging
10+
11+
from torchrec.distributed.logging_handlers import _log_handlers
12+
13+
TORCHREC_LOGGER_NAME = "torchrec"
14+
15+
_log_handlers.update({TORCHREC_LOGGER_NAME: logging.NullHandler()})

0 commit comments

Comments
 (0)