From cfd74fc6d64e5ac8fcae05874d1a7ecb79e2e327 Mon Sep 17 00:00:00 2001 From: Jeff Daily Date: Mon, 9 Oct 2023 18:14:27 +0000 Subject: [PATCH] revert file softmax.h --- onnxruntime/core/providers/rocm/math/softmax.h | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/onnxruntime/core/providers/rocm/math/softmax.h b/onnxruntime/core/providers/rocm/math/softmax.h index 0a5571bd57dde..49bfddad36b43 100644 --- a/onnxruntime/core/providers/rocm/math/softmax.h +++ b/onnxruntime/core/providers/rocm/math/softmax.h @@ -11,7 +11,7 @@ namespace rocm { using tunable::RocmTuningContext; -template +template Status SoftMaxComputeHelper( Stream* stream, const T* input, @@ -20,14 +20,14 @@ Status SoftMaxComputeHelper( int64_t axis, RocmTuningContext* tuning_ctx = nullptr); -template -Status dispatch_warpwise_softmax_forward(Stream* stream, output_t* dst, const input_t* src, - int softmax_elements, int softmax_elements_stride, int batch_count, +template +Status dispatch_warpwise_softmax_forward(Stream* stream, OutputT* dst, const InputT* src, int softmax_elements, + int softmax_elements_stride, int batch_count, RocmTuningContext* tuning_ctx = nullptr); -template -Status dispatch_blockwise_softmax_forward(Stream* stream, output_t* output, const input_t* input, - int softmax_elements, int input_stride, int output_stride, int batch_count, +template +Status dispatch_blockwise_softmax_forward(Stream* stream, OutputT* output, const InputT* input, int softmax_elements, + int input_stride, int output_stride, int batch_count, RocmTuningContext* tuning_ctx = nullptr); template