| # # 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 | |