- Fix to linking for GPU builds of softmax inference

test=develop

- Fix to linking for GPU builds of softmax inference
test=develop
03299ed4 · Jacek Czaja · 07563437 · 03299ed4
隐藏空白更改
内联并排

Showing with 4 addition and 0 deletion

paddle/fluid/operators/math/softmax.cu paddle/fluid/operators/math/softmax.cu +4 -0

未找到文件。
--- a/paddle/fluid/operators/math/softmax.cu
+++ b/paddle/fluid/operators/math/softmax.cu
@@ -100,8 +100,12 @@ template class SoftmaxGradCUDNNFunctor<platform::float16>;

 template class SoftmaxFunctor<platform::CUDADeviceContext, platform::float16,
                              false>;
+template class SoftmaxFunctor<platform::CUDADeviceContext, platform::float16,
+                              true>;
 template class SoftmaxFunctor<platform::CUDADeviceContext, float, false>;
 template class SoftmaxFunctor<platform::CUDADeviceContext, double, false>;
+template class SoftmaxFunctor<platform::CUDADeviceContext, float, true>;
+template class SoftmaxFunctor<platform::CUDADeviceContext, double, true>;
 template class SoftmaxGradFunctor<platform::CUDADeviceContext, float>;
 template class SoftmaxGradFunctor<platform::CUDADeviceContext, double>;
 template class SoftmaxGradFunctor<platform::CUDADeviceContext,