Skip to content
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

Greatly reduce the host memory usage and device memory usage during autocompare #50

Merged
merged 4 commits into from
Sep 29, 2024

Conversation

zhaoguochun1995
Copy link
Collaborator

@zhaoguochun1995 zhaoguochun1995 commented Sep 27, 2024

Greatly reduce the host memory usage and device memory usage during autocompare
大大减小自动精度对比时的host内存占用和device内存占用
支持hook自定义算子返回值
优化算子参数打印(支持同时打印cpu output和 cpu grad_inputs)
support hook custom operator output
and optimize operator parameter printing
截屏2024-09-29 下午6 02 30

截屏2024-09-29 上午11 59 38
截屏2024-09-29 下午2 44 10
截屏2024-09-29 下午4 34 15

@zhaoguochun1995 zhaoguochun1995 force-pushed the zgc/ditorch_reduce_autocompare_memory_usage branch from 578222c to 119867c Compare September 29, 2024 10:04
@zhaoguochun1995 zhaoguochun1995 force-pushed the zgc/ditorch_reduce_autocompare_memory_usage branch from 119867c to 41c130d Compare September 29, 2024 10:05
@yangbofun yangbofun merged commit 23761cb into main Sep 29, 2024
13 checks passed
@yangbofun yangbofun deleted the zgc/ditorch_reduce_autocompare_memory_usage branch September 29, 2024 10:50
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants