microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-04-20 06:49:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

wip

Browse Source
This commit is contained in:
Caio Rocha
2026-03-04 18:53:21 +00:00
parent 69565a2f32
commit 03fc782c42
1 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
python/mscclpp/language/internal/operations.py
View File

@@ -276,7 +276,8 @@ class SignalOperation(BaseOperation):
super().__init__(Instruction.relaxed_signal)
else:
super().__init__(Instruction.signal)
self.channel_ids = set(channels_ids)
self.channel_ids = channels_ids
self.check_channel_ids = set(channels_ids)
self.channel_type = channel_type
self.data_sync = data_sync
@@ -286,10 +287,10 @@ class SignalOperation(BaseOperation):
isinstance(other, SignalOperation)
and self.channel_type == other.channel_type
and self.name == other.name
and not self.channel_ids & other.channel_ids
and not self.check_channel_ids & other.check_channel_ids
):
fused_operation = SignalOperation(
channels_ids=self.channel_ids | other.channel_ids,
channels_ids=self.channel_ids + other.channel_ids,
channel_type=self.channel_type,
data_sync=self.data_sync | other.data_sync,
relaxed=(self.name == Instruction.relaxed_signal),
@@ -323,7 +324,8 @@ class WaitOperation(BaseOperation):
super().__init__(Instruction.relaxed_wait)
else:
super().__init__(Instruction.wait)
self.channel_ids = set(channels_ids)
self.channel_ids = channels_ids
self.check_channel_ids = set(channels_ids)
self.channel_type = channel_type
self.data_sync = data_sync
@@ -332,11 +334,11 @@ class WaitOperation(BaseOperation):
if (
isinstance(other, WaitOperation)
and self.name == other.name
and not self.channel_ids & other.channel_ids
and not self.check_channel_ids & other.check_channel_ids
and self.channel_type == other.channel_type
):
fused_operation = WaitOperation(
channels_ids=self.channel_ids | other.channel_ids,
channels_ids=self.channel_ids + other.channel_ids,
channel_type=self.channel_type,
data_sync=self.data_sync | other.data_sync,
relaxed=(self.name == Instruction.relaxed_wait),
@@ -408,7 +410,7 @@ class BarrierOperation(BaseOperation):
class FlushOperation(BaseOperation):
def __init__(self, channels_ids: List[int], channel_type: ChannelType, data_sync: SyncType = SyncType.none):
super().__init__(Instruction.flush)
self.channel_ids = set(channels_ids)
self.channel_ids = channels_ids
self.channel_type = channel_type
self.data_sync = data_sync
@@ -416,7 +418,7 @@ class FlushOperation(BaseOperation):
fused_operation = None
if isinstance(other, FlushOperation) and self.channel_type == other.channel_type:
fused_operation = FlushOperation(
channels_ids=self.channel_ids | other.channel_ids,
channels_ids=self.channel_ids + other.channel_ids,
channel_type=self.channel_type,
data_sync=self.data_sync | other.data_sync,
)