-
Notifications
You must be signed in to change notification settings - Fork 164
cambricon: update cambricon code based on master #1126
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
base: master
Are you sure you want to change the base?
cambricon: update cambricon code based on master #1126
Conversation
|
|
dee097c to
8c21ce7
Compare
| shift_amount = torch.randint( | ||
| 0, 8, shape, dtype=cur_dtype, device=self.device | ||
| shift_amount = torch.randint(0, 8, shape, dtype=cur_dtype, device="cpu").to( | ||
| self.device |
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.
Is there any difference between these two expressions?
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.
Since the randint function in our torch-mlu does not support specific data types (e.g., i16), we’ve made this temporary modification for now. Support for these types will be added in future.
53e6f62 to
20ca84b
Compare
|
lgtm |
20ca84b to
8380751
Compare
fdce5f1 to
63166a3
Compare
63166a3 to
bffbc93
Compare
PR Category
Operator
Type of Change
Performance Optimization
Description
update cambricon to master
Issue
Progress
Performance