Skip to content

Fix check_tensor to work with large models on Windows #5227

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged

Conversation

PatriceVignola
Copy link
Contributor

Description

Fix check_tensor to work with large models

Motivation and Context

Calling check_tensor on large 7B+ LLM models currently fails because the size doesn't fit in int32, so using _wstat64 instead fixes the problem.

@PatriceVignola PatriceVignola requested a review from a team as a code owner May 14, 2023 03:47
@jcwchen jcwchen added the module: checker onnx.checker label May 14, 2023
Copy link
Member

@jcwchen jcwchen left a comment

Choose a reason for hiding this comment

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

Thank you for enhancing checker! Please follow the steps to fix DCO issue before merge.

@xadupre
Copy link
Contributor

xadupre commented May 17, 2023

You need to sign your commit (git commit -s ...) to make the DCO pass or follow the instructions after clicking on details.

Signed-off-by: Patrice Vignola <vignola.patrice@gmail.com>
@PatriceVignola PatriceVignola force-pushed the user/pavignol/fix-check-tensor-large-models branch from d0ac30a to 6f4051a Compare May 18, 2023 08:51
@PatriceVignola PatriceVignola requested a review from jcwchen May 18, 2023 08:52
@jcwchen jcwchen added the run release CIs Use this label to trigger release tests in CI label May 18, 2023
@jcwchen jcwchen merged commit 60cebc3 into onnx:main Jun 5, 2023
adityagoel4512 pushed a commit to adityagoel4512/onnx that referenced this pull request Jul 28, 2023
Signed-off-by: Patrice Vignola <vignola.patrice@gmail.com>
Co-authored-by: Chun-Wei Chen <jacky82226@gmail.com>
Co-authored-by: Xavier Dupré <xadupre@users.noreply.github.com>
Signed-off-by: Aditya Goel <agoel4512@gmail.com>
@abock abock mentioned this pull request Aug 3, 2023
5 tasks
yuanyao-nv pushed a commit to yuanyao-nv/onnx that referenced this pull request Aug 7, 2023
Signed-off-by: Patrice Vignola <vignola.patrice@gmail.com>
Co-authored-by: Chun-Wei Chen <jacky82226@gmail.com>
Co-authored-by: Xavier Dupré <xadupre@users.noreply.github.com>
yuanyao-nv pushed a commit to yuanyao-nv/onnx that referenced this pull request Aug 7, 2023
Signed-off-by: Patrice Vignola <vignola.patrice@gmail.com>
Co-authored-by: Chun-Wei Chen <jacky82226@gmail.com>
Co-authored-by: Xavier Dupré <xadupre@users.noreply.github.com>
Signed-off-by: Yuan Yao <yuanyao@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
module: checker onnx.checker run release CIs Use this label to trigger release tests in CI
Projects
Status: Done
Development

Successfully merging this pull request may close these issues.

4 participants