saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix build error for generation debug code (#12801)

Browse source 
fix dumping code
This commit is contained in:
Tianlei Wu 2022-08-31 13:53:01 -07:00 committed by GitHub
parent a48b115540
commit 7e1e0f5c9e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/contrib_ops/cpu/transformers/generation_device_helper.cc
View file

@ -468,8 +468,8 @@ Status GreedySearchProcessLogits(
topk_indices));
#ifdef DEBUG_GENERATION
dumper->Print("topk_scores", *(topk_scores.get()));
dumper->Print("topk_indices", *(topk_indices.get()));
dumper->Print("topk_scores", topk_scores);
dumper->Print("topk_indices", topk_indices);
#endif
gsl::span<const int64_t> next_token_indices = topk_indices.DataAsSpan<int64_t>();

4
onnxruntime/contrib_ops/cuda/transformers/generation_device_helper.cc
View file

@ -520,8 +520,8 @@ Status GreedySearchProcessLogits(
*topk_scores, *topk_indices));
#ifdef DEBUG_GENERATION
dumper->Print("topk_scores", topk_scores);
dumper->Print("topk_indices", topk_indices);
dumper->Print("topk_scores", *(topk_scores.get()));
dumper->Print("topk_indices", *(topk_indices.get()));
#endif
const int64_t* next_token_indices = topk_indices->Data<int64_t>();