コード例 #1
0
ファイル: extern.py プロジェクト: zzzDavid/heterocl
    def mutate_Load(self, node):
        buffer_var = self.mutate(node.buffer_var)
        index = self.mutate(node.index)
        index = util.CastRemover().mutate(index)
        self.record_index(buffer_var, index)

        predicate = self.mutate(node.predicate)
        return _make.Load(node.dtype, buffer_var, index, predicate)
コード例 #2
0
ファイル: extern.py プロジェクト: zzzDavid/heterocl
    def mutate_Store(self, node):
        buffer_var = self.mutate(node.buffer_var)
        index = self.mutate(node.index)
        index = util.CastRemover().mutate(index)
        self.record_index(buffer_var, index)

        value = self.mutate(node.value)
        predicate = self.mutate(node.predicate)
        return _make.Store(buffer_var, value, index, predicate)