__init__.py 309 Bytes
Newer Older
1
# Copyright (c) 2022-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2
3
4
5
6
7
8
9
10
11
#
# See LICENSE for license information.

"""Top level package for numerical debugging."""

try:
    from . import pytorch
    from .pytorch.debug_state import set_weight_tensor_tp_group_reduce
except ImportError as e:
    pass