From 533fe37cbd6be402e0975e35ac7c26fe3dde891b Mon Sep 17 00:00:00 2001 From: Yulong Wang <7679871+fs-eire@users.noreply.github.com> Date: Mon, 19 Dec 2022 16:49:21 -0800 Subject: [PATCH] fix build break in transformer debug dump (#14009) ### Description Fix build break in transformer debug dump introduced in #13954. --- onnxruntime/contrib_ops/cpu/transformers/beam_search_impl_gpt.h | 2 +- .../contrib_ops/cpu/transformers/greedy_search_impl_gpt.h | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/onnxruntime/contrib_ops/cpu/transformers/beam_search_impl_gpt.h b/onnxruntime/contrib_ops/cpu/transformers/beam_search_impl_gpt.h index d72a150091..46697a202d 100644 --- a/onnxruntime/contrib_ops/cpu/transformers/beam_search_impl_gpt.h +++ b/onnxruntime/contrib_ops/cpu/transformers/beam_search_impl_gpt.h @@ -251,7 +251,7 @@ Status BeamSearchGpt::Execute(const FeedsFetchesManager* init_run_feeds_fetch if (iteration_counter++ == 0 && init_run_decoder_session_state_ != nullptr) { #ifdef DEBUG_NODE_INPUTS_OUTPUTS - const_cast(this->init_run_decoder_session_state_).IncrementGraphExecutionCounter(); + const_cast(this->init_run_decoder_session_state_)->IncrementGraphExecutionCounter(); #endif status = utils::ExecuteSubgraph(*init_run_decoder_session_state_, *init_run_feeds_fetches_manager, diff --git a/onnxruntime/contrib_ops/cpu/transformers/greedy_search_impl_gpt.h b/onnxruntime/contrib_ops/cpu/transformers/greedy_search_impl_gpt.h index 22d41c9e19..cf12fe6ba1 100644 --- a/onnxruntime/contrib_ops/cpu/transformers/greedy_search_impl_gpt.h +++ b/onnxruntime/contrib_ops/cpu/transformers/greedy_search_impl_gpt.h @@ -242,7 +242,7 @@ Status GreedySearchGpt::Execute(const FeedsFetchesManager* init_run_feeds_fet if (iteration_counter++ == 0 && init_run_decoder_session_state_ != nullptr) { #ifdef DEBUG_NODE_INPUTS_OUTPUTS - const_cast(this->init_run_decoder_session_state_).IncrementGraphExecutionCounter(); + const_cast(this->init_run_decoder_session_state_)->IncrementGraphExecutionCounter(); #endif status = utils::ExecuteSubgraph(*init_run_decoder_session_state_, *init_run_feeds_fetches_manager,