File tree Expand file tree Collapse file tree 1 file changed +4
-0
lines changed
paddle/fluid/operators/math Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Original file line number Diff line number Diff line change @@ -100,8 +100,12 @@ template class SoftmaxGradCUDNNFunctor<platform::float16>;
100
100
101
101
template class SoftmaxFunctor <platform::CUDADeviceContext, platform::float16,
102
102
false >;
103
+ template class SoftmaxFunctor <platform::CUDADeviceContext, platform::float16,
104
+ true >;
103
105
template class SoftmaxFunctor <platform::CUDADeviceContext, float , false >;
104
106
template class SoftmaxFunctor <platform::CUDADeviceContext, double , false >;
107
+ template class SoftmaxFunctor <platform::CUDADeviceContext, float , true >;
108
+ template class SoftmaxFunctor <platform::CUDADeviceContext, double , true >;
105
109
template class SoftmaxGradFunctor <platform::CUDADeviceContext, float >;
106
110
template class SoftmaxGradFunctor <platform::CUDADeviceContext, double >;
107
111
template class SoftmaxGradFunctor <platform::CUDADeviceContext,
You can’t perform that action at this time.
0 commit comments