GitLab

Menu

Projects Groups Snippets

Loading...
Help
Sign in / Register

A apex
Project information
- Project information
- Activity
- Members
Repository
- Repository
- Files
- Commits
- Branches
- Tags
- Contributors
- Graph
- Compare
Jira
- Jira
CI/CD
- CI/CD
- Pipelines
- Jobs
- Schedules
Deployments
- Deployments
- Environments
- Releases
Monitor
- Monitor
- Metrics
Packages & Registries
- Packages & Registries
- Package Registry
- Infrastructure Registry
Analytics
- Analytics
- Value stream
- CI/CD
- Repository
Activity
Graph
Jobs
Commits

Collapse sidebar

mirrors
ROCmSoftwarePlatform
apex
Repository
Branches

Overview
Active
Stale
All

dev/athitten/multihead_attn merged

c3ec9351 · apex definition of macro conflicts with pytorch macro WARP_SHFL_XOR · Oct 22, 2021

Compare

Download source code

zip tar.gz tar.bz2 tar
mlp-debug merged

964e61f1 · Enable MLP unit tests on ROCm · Oct 27, 2021

Compare

Download source code

zip tar.gz tar.bz2 tar
dev/hubertlu/multihead_attn merged

62f06964 · Update setup.py · Nov 03, 2021

Compare

Download source code

zip tar.gz tar.bz2 tar
IFU-master-2021-10-15-multihead_attn

adf3b27c · Fix some merge conflicts · Nov 03, 2021

Compare

Download source code

zip tar.gz tar.bz2 tar
IFU-master-2021-11-09

309363f5 · Replace THCublasCheck with TORCH_CUDABLAS_CHECK · Nov 10, 2021

Compare

Download source code

zip tar.gz tar.bz2 tar
dev/athitten/distributed_lamb merged

f3868524 · Enable Distributed FusedLAMB · Nov 19, 2021

Compare

Download source code

zip tar.gz tar.bz2 tar
dev/hubertlu/unit_tests merged

2228f1bf · Merge remote-tracking branch 'origin/master' into dev/hubertlu/unit_tests · Dec 03, 2021

Compare

Download source code

zip tar.gz tar.bz2 tar
enable_all_supported_extensions

b4629e2e · Need to clear all cmdline args so setup.py doesn't complain · Dec 03, 2021

Compare

Download source code

zip tar.gz tar.bz2 tar
IFU-master-2021-10-15 merged

fec3141c · Replace THCudaCheck with C10_CUDA_CHECK · Dec 07, 2021

Compare

Download source code

zip tar.gz tar.bz2 tar
dev/hubertlu/skip_unittest

5fbb3515 · Modify the test skipping messages · Dec 14, 2021

Compare

Download source code

zip tar.gz tar.bz2 tar
IFU-master-2021-12-08 merged

68364b49 · Conditionally define autocast_dtypes for different torch versions · Dec 15, 2021

Compare

Download source code

zip tar.gz tar.bz2 tar
gbn-channels-last

0435d4c4 · Fix the memory-format API · Dec 15, 2021

Compare

Download source code

zip tar.gz tar.bz2 tar
gbn-pytorch-channels-last

4c29c199 · Add support for memory_format=torch.channels_last in GBN · Dec 15, 2021

Compare

Download source code

zip tar.gz tar.bz2 tar
fused_layer_norm_optim

fea95106 · Optimize cuComputePartGradGammaBeta for AMD GPUs · Jan 08, 2022

Compare

Download source code

zip tar.gz tar.bz2 tar
pru_319886

a2ed1b0c · Fix for compilation failure · Jan 26, 2022

Compare

Download source code

zip tar.gz tar.bz2 tar
cherry_pick_b2fdf9c

e3f2ecb7 · Cherry-pick b2fdf9c4 from upstream Apex and resolve conflicts · Jan 27, 2022

Compare

Download source code

zip tar.gz tar.bz2 tar
dev/athitten/denorm_fix_bwd_gemms

046a4291 · Add denorm fix for multihead_attn bwd rocblas calls · Jan 29, 2022

Compare

Download source code

zip tar.gz tar.bz2 tar
dev/hubertlu/gbn_optim

8aeffee3 · Revert rectified, stg, and stg_stream · Feb 26, 2022

Compare

Download source code

zip tar.gz tar.bz2 tar
dev/hubertlu/gbn_optim_new

245be05d · Re-enable gbn module in Apex for ROCm · Mar 01, 2022

Compare

Download source code

zip tar.gz tar.bz2 tar
dev/athitten/mha_roblas_alt_impl_flag

2be68b18 · Use ifdef for rocblas_gemm_flags_fp16_alt_impl to target at various AMD hardware · Mar 17, 2022

Compare

Download source code

zip tar.gz tar.bz2 tar

Prev
1
…
3
4
5
6
7
8
9
10
Next

Download source code

Download source code

Download source code

Download source code

Download source code

Download source code

Download source code

Download source code

Download source code

Download source code

Download source code

Download source code

Download source code

Download source code

Download source code

Download source code

Download source code

Download source code

Download source code

Download source code