Skip to content

Infernce speed good but one thing left to optimze gpu sync - #134

Merged
E53klasky merged 10 commits into
masterfrom
infernce-speed
Jul 27, 2026
Merged

Infernce speed good but one thing left to optimze gpu sync#134
E53klasky merged 10 commits into
masterfrom
infernce-speed

Conversation

@E53klasky

Copy link
Copy Markdown
Member

No description provided.

@E53klasky E53klasky left a comment

Copy link
Copy Markdown
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

fine

@E53klasky

Copy link
Copy Markdown
Member Author

gpu syncs here; torch::Tensor raw_batched_input =
batch_input_buf.narrow(0, 0, cur_count).clone();
torch::Tensor batched_indexes =
batch_index_buf.narrow(0, 0, cur_count).clone().to(device_);

@E53klasky

Copy link
Copy Markdown
Member Author

fix the data filtering

@E53klasky
E53klasky merged commit 83c83ff into master Jul 27, 2026
8 checks passed
@E53klasky
E53klasky deleted the infernce-speed branch July 27, 2026 11:01
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant