ROCm / apex Public

forked from NVIDIA/apex

Notifications You must be signed in to change notification settings
Fork 19
Star 19

Code
Issues 13
Pull requests 6
Actions
Projects
Wiki
Security
Insights

Additional navigation options

Code
Issues
Pull requests
Actions
Projects
Wiki
Security
Insights

Pull requests: ROCm/apex

Labels 10 Milestones 0

New pull request New

6 Open 114 Closed

Author

Filter by author

Label

Filter by label

Use alt + click/return to exclude labels

or ⇧ + click/return for logical OR

Projects

Filter by project

Milestones

Filter by milestone

Reviews

Filter by reviews

No reviews Review required Approved review Changes requested

Assignee

Filter by who’s assigned

Assigned to nobody

Sort

Sort by

Newest Oldest Most commented Least commented Recently updated Least recently updated Best match

Most reactions

Pull requests list

setup: check if torch.cuda is available before calling cuda prop

#140 opened Nov 7, 2024 by jagadish-amd

Loading…

Heyi fused grad accumulation

#138 opened Sep 26, 2024 by eliotwang

Loading…

Add setting of env flag when apex is turned on

#114 opened Aug 23, 2023 by alugorey

Loading…

Support all the softmax extensions and cherry-pick transformer-related commits

#101 opened Dec 30, 2022 by hubertlu-tw

Loading…

Enable --fast_layer_norm for ROCm

#94 opened Sep 9, 2022 by hubertlu-tw

Loading…

Cherry-pick fused_adam_cuda related patches and add its unit tests to the ROCm extension test script

#93 opened Sep 9, 2022 by hubertlu-tw

Loading…

ProTip! Type g p on any issue or pull request to go back to the pull request listing page.

Provide feedback

Saved searches

Use saved searches to filter your results more quickly