DeepSpeed
Fix the FP6 kernels compilation problem on non-Ampere GPUs.
#5333

Merged

Fix the FP6 kernels compilation problem on non-Ampere GPUs. #5333

mrwyattii merged 7 commits into deepspeedai:master from JamesTheZ:features/fp6_compile_err

JamesTheZ requested a review from

mrwyattii 1 year ago

JamesTheZ requested a review from

awan-10 1 year ago

JamesTheZ requested a review from

arashb 1 year ago

JamesTheZ requested a review from

loadams 1 year ago

JamesTheZ requested a review from

tjruwase 1 year ago

JamesTheZ force pushed from fcc5653c to 505e5da0 1 year ago

loadams commented on 2024-04-01

JamesTheZ force pushed from f166df80 to d21ff568 1 year ago

Refine the guards of FP6 kernels compilation.

976c2206

Fix rocm checking.

06b6358b

Add the header file ignored due to .gitignore

3ecd9bd7

Remove unnecessary guards.

72c86fde

Reduce the tests of FP6 kernel.

f7904be3

JamesTheZ force pushed from 81a63d5a to f7904be3 1 year ago

Merge branch 'master' into features/fp6_compile_err

88bd9928

mrwyattii approved these changes on 2024-04-15

Merge branch 'master' into features/fp6_compile_err

98703121

mrwyattii enabled auto-merge 1 year ago

mrwyattii merged e3d873a0 into master 1 year ago

Reviewers

mrwyattii

loadams

awan-10

arashb

tjruwase

Assignees

No one assigned

Labels

None yet

Milestone

No milestone