diff --git a/onnxruntime/core/providers/rocm/math/softmax.cc b/onnxruntime/core/providers/rocm/math/softmax.cc index bae1c000ddfcc..64edbc1558d1e 100644 --- a/onnxruntime/core/providers/rocm/math/softmax.cc +++ b/onnxruntime/core/providers/rocm/math/softmax.cc @@ -11,7 +11,7 @@ namespace onnxruntime { namespace rocm { -template +template Status SoftMaxComputeHelper( Stream* stream, const T* X, @@ -30,11 +30,11 @@ Status SoftMaxComputeHelper( if (D <= 1024 && D * sizeof(T) <= 4096) { return dispatch_warpwise_softmax_forward< - HipT_IN, HipT_OUT, AccumulationType_t, is_log_softmax>( + HipT_IN, HipT_OUT, AccumulationType_t, IsLogSoftmax>( stream, Y_data, X_data, gsl::narrow_cast(D), gsl::narrow_cast(D), gsl::narrow_cast(N), tuning_ctx); } - return dispatch_blockwise_softmax_forward, is_log_softmax>( + return dispatch_blockwise_softmax_forward, IsLogSoftmax>( stream, Y_data, X_data, gsl::narrow_cast(D), gsl::narrow_cast(D), gsl::narrow_cast(D), gsl::narrow_cast(N), tuning_ctx); }