-
Notifications
You must be signed in to change notification settings - Fork 56
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
转换规则 No. 141/142 #180
转换规则 No. 141/142 #180
Conversation
Thanks for your contribution! |
|
||
|
||
def test_case_1(): | ||
pytorch_code = textwrap.dedent( |
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.
这种cuda类型的api先在本地跑看是否单测能够通过,截图可以贴到下面。CI没有GPU,无法跑这个单测,可以用torch.cuda is_available()来绕过。
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.
已修改
tests/test_cuda_nvtx_range_push.py
Outdated
pytorch_code = textwrap.dedent( | ||
""" | ||
import torch | ||
torch.cuda.nvtx.range_push("msg") |
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.
这种cuda类型的api先在本地跑看是否单测能够通过,截图可以贴到下面。CI没有GPU,无法跑这个单测,可以用torch.cuda is_available()来绕过。
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.
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.
已修改
PR Docs
#112
PR APIs