softmax_op.cu

#include "paddle/framework/op_registry.h"
#include "paddle/operators/softmax_op.h"

REGISTER_OP_GPU_KERNEL(
    softmax, paddle::operators::SoftmaxKernel<paddle::platform::GPUPlace, float>);