Skip to content

[CUDA] BF16 support - #7014

Merged
jcf94 merged 2 commits into
apache:mainfrom
hgt312:cuda_bfp16
Mar 6, 2021
Merged

[CUDA] BF16 support#7014
jcf94 merged 2 commits into
apache:mainfrom
hgt312:cuda_bfp16

Conversation

@hgt312

@hgt312 hgt312 commented Dec 2, 2020

Copy link
Copy Markdown
Contributor

@tqchen

tqchen commented Mar 3, 2021

Copy link
Copy Markdown
Member

@yzhliu @hgt312 can we please followup on this PR?

@jcf94 would be great if you can help to manage this PR (also cc @vinx13 @masahi)

@jcf94 jcf94 left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Plz rebase this branch

Comment thread tests/python/unittest/test_target_codegen_cuda.py Outdated
@jcf94

jcf94 commented Mar 4, 2021

Copy link
Copy Markdown
Member

@yzhliu @hgt312 can we please followup on this PR?

@jcf94 would be great if you can help to manage this PR (also cc @vinx13 @masahi)

@tqchen Thanks! I'd love to help.

If @hgt312 don't have time to finish these remaining issues, I can help clean them this weekend.

@hgt312
hgt312 requested a review from jcf94 March 4, 2021 10:34

@jcf94 jcf94 left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks! @hgt312

@jcf94
jcf94 merged commit 8aa2a7c into apache:main Mar 6, 2021
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request May 6, 2021
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request May 11, 2021
@hgt312
hgt312 deleted the cuda_bfp16 branch December 12, 2021 01:38
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