saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-04 23:59:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix build break in transformer debug dump (#14009)

Browse source 
### Description

Fix build break in transformer debug dump introduced in #13954.
This commit is contained in:
Yulong Wang 2022-12-19 16:49:21 -08:00 committed by GitHub
parent fc2a6db573
commit 533fe37cbd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/contrib_ops/cpu/transformers/beam_search_impl_gpt.h
View file

@ -251,7 +251,7 @@ Status BeamSearchGpt<T>::Execute(const FeedsFetchesManager* init_run_feeds_fetch
if (iteration_counter++ == 0 &&
init_run_decoder_session_state_ != nullptr) {
#ifdef DEBUG_NODE_INPUTS_OUTPUTS
const_cast<SessionState&>(this->init_run_decoder_session_state_).IncrementGraphExecutionCounter();
const_cast<SessionState*>(this->init_run_decoder_session_state_)->IncrementGraphExecutionCounter();
#endif
status = utils::ExecuteSubgraph(*init_run_decoder_session_state_,
*init_run_feeds_fetches_manager,

2
onnxruntime/contrib_ops/cpu/transformers/greedy_search_impl_gpt.h
View file

@ -242,7 +242,7 @@ Status GreedySearchGpt<T>::Execute(const FeedsFetchesManager* init_run_feeds_fet
if (iteration_counter++ == 0 &&
init_run_decoder_session_state_ != nullptr) {
#ifdef DEBUG_NODE_INPUTS_OUTPUTS
const_cast<SessionState&>(this->init_run_decoder_session_state_).IncrementGraphExecutionCounter();
const_cast<SessionState*>(this->init_run_decoder_session_state_)->IncrementGraphExecutionCounter();
#endif
status = utils::ExecuteSubgraph(*init_run_decoder_session_state_,
*init_run_feeds_fetches_manager,