Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

[BACKPORT][FEATURE]Support for fp16 in SpM x DnsM on GPU (#18930) #19074

Merged
merged 1 commit into from
Sep 3, 2020

Conversation

access2rohit
Copy link
Contributor

@access2rohit access2rohit commented Sep 2, 2020

Description

  • Support for fp16 in SpGeMM

  • adding test for GPU spmm

Backported #18930

Checklist

Essentials

  • PR's title starts with a category (e.g. [BUGFIX], [MODEL], [TUTORIAL], [FEATURE], [DOC], etc)
  • Changes are complete (i.e. I finished coding on this PR)
  • All changes have test coverage
  • Code is well-documented

* Support for fp16 in SpGeMM

* adding test for GPU spmm

Co-authored-by: Rohit Kumar Srivastava <srivastava.141@buckeyemail.osu.edu>
@mxnet-bot
Copy link

Hey @access2rohit , Thanks for submitting the PR
All tests are already queued to run once. If tests fail, you can trigger one or more tests again with the following commands:

  • To trigger all jobs: @mxnet-bot run ci [all]
  • To trigger specific jobs: @mxnet-bot run ci [job1, job2]

CI supported jobs: [miscellaneous, website, clang, edge, windows-gpu, windows-cpu, centos-cpu, sanity, unix-gpu, unix-cpu, centos-gpu]


Note:
Only following 3 categories can trigger CI :PR Author, MXNet Committer, Jenkins Admin.
All CI tests must pass before the PR can be merged.

@samskalicky samskalicky merged commit 1a398e5 into apache:v1.x Sep 3, 2020
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants