Skip to content

Fix FP16 support on ROCm platforms#2366

Merged
benson31 merged 2 commits into
LBANN:developfrom
benson31:fix-rocm-fp16
Oct 30, 2023
Merged

Fix FP16 support on ROCm platforms#2366
benson31 merged 2 commits into
LBANN:developfrom
benson31:fix-rocm-fp16

Conversation

@benson31

Copy link
Copy Markdown
Collaborator

This requires llnl/Elemental#109.

This does NOT impact DistConv codepaths -- fp16 is still not supported in DistConv on any platform.

@ndryden will likely care about this for testing AMP (#2359).

@benson31 benson31 merged commit 2616203 into LBANN:develop Oct 30, 2023
@benson31 benson31 deleted the fix-rocm-fp16 branch October 30, 2023 17:35
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants