-
Notifications
You must be signed in to change notification settings - Fork 57
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
【Hackathon 5th No.45】convert torch.cuda.comm.broadcast, torch.cuda.device -part #359
Conversation
Thanks for your contribution! |
@zhwesky2010 没有找到这个耶:torch.nn.modules.batchnorm._NormBase |
tests/test_cuda_comm_broadcast.py
Outdated
obj = APIBase("torch.cuda.comm.broadcast") | ||
|
||
|
||
def test_case_1(): |
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.
@longranger2 你可能需要本地测试一下,因为目前CI上是CPU的torch
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.
以实际代码运行为判断标准,不要以文档为准 |
@longranger2 单侧还没有过 |
PR Docs
PaddlePaddle/docs#6217
PR APIs