From 2afe507a5f46c73bcf6b706a8a02a4133242c155 Mon Sep 17 00:00:00 2001 From: Jeff Daily Date: Mon, 9 Oct 2023 18:31:17 +0000 Subject: [PATCH] partial revert file softmax.cc --- onnxruntime/core/providers/rocm/math/softmax.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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); }