Skip to content

Conversation

@chenmiao1919
Copy link
Contributor

PR Category

Operator

Type of Change

Performance Optimization

Description

update cambricon to master

Issue

Progress

  • Change is properly reviewed (1 reviewer required, 2 recommended).
  • Change is responded to an issue.
  • Change is fully covered by a UT.

Performance

@CLAassistant
Copy link

CLA assistant check
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
You have signed the CLA already but the status is still pending? Let us recheck it.

@chenmiao1919 chenmiao1919 force-pushed the cambricon_merge_to_master_q4 branch 3 times, most recently from dee097c to 8c21ce7 Compare November 28, 2025 12:49
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
Copy link
Collaborator

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?

Copy link
Contributor Author

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.

@chenmiao1919 chenmiao1919 force-pushed the cambricon_merge_to_master_q4 branch 2 times, most recently from 53e6f62 to 20ca84b Compare December 2, 2025 02:50
@kiddyjinjin
Copy link
Collaborator

lgtm

kiddyjinjin
kiddyjinjin previously approved these changes Dec 2, 2025
@chenmiao1919 chenmiao1919 force-pushed the cambricon_merge_to_master_q4 branch from 20ca84b to 8380751 Compare December 3, 2025 01:58
@chenmiao1919 chenmiao1919 force-pushed the cambricon_merge_to_master_q4 branch 9 times, most recently from fdce5f1 to 63166a3 Compare December 5, 2025 07:20
@chenmiao1919 chenmiao1919 force-pushed the cambricon_merge_to_master_q4 branch from 63166a3 to bffbc93 Compare December 5, 2025 10:15
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.

3 participants