|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#include "filtered_lrelu.cu" |
|
|
|
|
|
|
|
|
|
template filtered_lrelu_kernel_spec choose_filtered_lrelu_kernel<c10::Half, int32_t, true, false>(const filtered_lrelu_kernel_params& p, int sharedKB); |
|
template filtered_lrelu_kernel_spec choose_filtered_lrelu_kernel<float, int32_t, true, false>(const filtered_lrelu_kernel_params& p, int sharedKB); |
|
|
|
|
|
template filtered_lrelu_kernel_spec choose_filtered_lrelu_kernel<c10::Half, int64_t, true, false>(const filtered_lrelu_kernel_params& p, int sharedKB); |
|
template filtered_lrelu_kernel_spec choose_filtered_lrelu_kernel<float, int64_t, true, false>(const filtered_lrelu_kernel_params& p, int sharedKB); |
|
|
|
|
|
template void* choose_filtered_lrelu_act_kernel<c10::Half, true, false>(void); |
|
template void* choose_filtered_lrelu_act_kernel<float, true, false>(void); |
|
template void* choose_filtered_lrelu_act_kernel<double, true, false>(void); |
|
|
|
|
|
template cudaError_t copy_filters<true, false>(cudaStream_t stream); |
|
|