|
|
|
* SPDX-FileCopyrightText: Copyright (c) 2021-2022 NVIDIA CORPORATION & AFFILIATES. All rights reserved. |
|
* SPDX-License-Identifier: LicenseRef-NvidiaProprietary |
|
* |
|
* NVIDIA CORPORATION, its affiliates and licensors retain all intellectual |
|
* property and proprietary rights in and to this material, related |
|
* documentation and any modifications thereto. Any use, reproduction, |
|
* disclosure or distribution of this material and related documentation |
|
* without an express license agreement from NVIDIA CORPORATION or |
|
* its affiliates is strictly prohibited. |
|
*/ |
|
|
|
#include "filtered_lrelu.cu" |
|
|
|
|
|
|
|
|
|
template filtered_lrelu_kernel_spec choose_filtered_lrelu_kernel<c10::Half, int32_t, false, false>(const filtered_lrelu_kernel_params& p, int sharedKB); |
|
template filtered_lrelu_kernel_spec choose_filtered_lrelu_kernel<float, int32_t, false, false>(const filtered_lrelu_kernel_params& p, int sharedKB); |
|
|
|
|
|
template filtered_lrelu_kernel_spec choose_filtered_lrelu_kernel<c10::Half, int64_t, false, false>(const filtered_lrelu_kernel_params& p, int sharedKB); |
|
template filtered_lrelu_kernel_spec choose_filtered_lrelu_kernel<float, int64_t, false, false>(const filtered_lrelu_kernel_params& p, int sharedKB); |
|
|
|
|
|
template void* choose_filtered_lrelu_act_kernel<c10::Half, false, false>(void); |
|
template void* choose_filtered_lrelu_act_kernel<float, false, false>(void); |
|
template void* choose_filtered_lrelu_act_kernel<double, false, false>(void); |
|
|
|
|
|
template cudaError_t copy_filters<false, false>(cudaStream_t stream); |
|
|