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