-
Notifications
You must be signed in to change notification settings - Fork 6.7k
[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] |
mseth10
left a comment
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