lib/torch_utils/ops/__init__.py

# # Copyright (c) 2021, NVIDIA CORPORATION.  All rights reserved.
# #
# # NVIDIA CORPORATION and its licensors retain all intellectual property
# # and proprietary rights in and to this software, related documentation
# # and any modifications thereto.  Any use, reproduction, disclosure or
# # distribution of this software and related documentation without an express
# # license agreement from NVIDIA CORPORATION is strictly prohibited.

# # empty
# try:
#     from .fused_act import FusedLeakyReLU, fused_leaky_relu
#     # from .upfirdn2d import upfirdn2d

#     print('Using custom CUDA kernels')
# except Exception as e:
#     print(str(e))
#     print('There was something wrong with the CUDA kernels')
#     print('Reverting to native PyTorch implementation')
#     from .native_ops import FusedLeakyReLU, fused_leaky_relu