Skip to content

perf: reduce redundant CUDA initialization and synchronization#190

Open
chen2021673 wants to merge 1 commit into
masterfrom
remove_fill
Open

perf: reduce redundant CUDA initialization and synchronization#190
chen2021673 wants to merge 1 commit into
masterfrom
remove_fill

Conversation

@chen2021673

@chen2021673 chen2021673 commented Jul 15, 2026

Copy link
Copy Markdown
Contributor

Summary

  • defer loss host synchronization until after backward in GPT-2, Llama3, MNIST, and pipeline training
  • skip zero initialization where CUDA kernels or cuBLAS fully overwrite output tensors
  • add numerical CPU and CUDA coverage for the affected operators

Test

image image

- defer loss host synchronization until after backward across training paths
- skip zero fills when CUDA kernels or cuBLAS fully overwrite outputs
- retain and document initialization required by partial-write paths
- add CPU and CUDA numerical tests for affected operators
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