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)
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)