-
Notifications
You must be signed in to change notification settings - Fork 6.8k
[WIP][DO NOT MERGE] Add large tensor test for linalg_dgemm2 #18718
Conversation
Hey @access2rohit , Thanks for submitting the PR
CI supported jobs: [windows-cpu, unix-cpu, centos-gpu, clang, centos-cpu, unix-gpu, windows-gpu, edge, miscellaneous, sanity, website] Note: |
@ChaiBapchya @mseth10 can you review ? |
@mxnet-label-bot add [pr-awaiting-review] |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
we are deprecating ndarray in 2.0 soon. we should start migrating tests to numpy arrays instead |
Changed it to v1.x |
@access2rohit as linked in this PR: #18784 I already added forward & backward test for linalg.gemm2 in v1.x |
Description
checks for correctness for linalg_gemm2 operator for Large Matrices
Checklist
Essentials
Please feel free to remove inapplicable items for your PR.
Test Results