Skip to content

Commit 03299ed

Browse files
committed
- Fix to linking for GPU builds of softmax inference
test=develop
1 parent 0756343 commit 03299ed

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

paddle/fluid/operators/math/softmax.cu

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -100,8 +100,12 @@ template class SoftmaxGradCUDNNFunctor<platform::float16>;
100100

101101
template class SoftmaxFunctor<platform::CUDADeviceContext, platform::float16,
102102
false>;
103+
template class SoftmaxFunctor<platform::CUDADeviceContext, platform::float16,
104+
true>;
103105
template class SoftmaxFunctor<platform::CUDADeviceContext, float, false>;
104106
template class SoftmaxFunctor<platform::CUDADeviceContext, double, false>;
107+
template class SoftmaxFunctor<platform::CUDADeviceContext, float, true>;
108+
template class SoftmaxFunctor<platform::CUDADeviceContext, double, true>;
105109
template class SoftmaxGradFunctor<platform::CUDADeviceContext, float>;
106110
template class SoftmaxGradFunctor<platform::CUDADeviceContext, double>;
107111
template class SoftmaxGradFunctor<platform::CUDADeviceContext,

0 commit comments

Comments
 (0)