srenblad/artiq
1
0
Fork 0
forked from M-Labs/artiq
Code Pull Requests Activity

compiler: Fix method name typo [nfc]

Browse Source
This commit is contained in:
David Nadlinger 2020-08-02 20:25:07 +01:00
parent 78afa2ea8e
commit 0da4a61d99
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
artiq/compiler/transforms/artiq_ir_generator.py
View File

@ -1398,7 +1398,7 @@ class ARTIQIRGenerator(algorithm.Visitor):
operand = self.visit(node.operand) operand = self.visit(node.operand)
if builtins.is_array(operand.type): if builtins.is_array(operand.type):
shape = self.append(ir.GetAttr(operand, "shape")) shape = self.append(ir.GetAttr(operand, "shape"))
result = self._alloate_new_array(node.type.find()["elt"], shape) result = self._allocate_new_array(node.type.find()["elt"], shape)
func = self._get_array_unaryop("USub", make_sub, node.type, operand.type) func = self._get_array_unaryop("USub", make_sub, node.type, operand.type)
self._invoke_arrayop(func, [result, operand]) self._invoke_arrayop(func, [result, operand])
return result return result
@ -1418,7 +1418,7 @@ class ARTIQIRGenerator(algorithm.Visitor):
if builtins.is_array(node.type): if builtins.is_array(node.type):
result_elt = node.type.find()["elt"] result_elt = node.type.find()["elt"]
shape = self.append(ir.GetAttr(value, "shape")) shape = self.append(ir.GetAttr(value, "shape"))
result = self._alloate_new_array(result_elt, shape) result = self._allocate_new_array(result_elt, shape)
func = self._get_array_unaryop("Coerce", func = self._get_array_unaryop("Coerce",
lambda v: ir.Coerce(v, result_elt), lambda v: ir.Coerce(v, result_elt),
node.type, value.type) node.type, value.type)
@ -1438,7 +1438,7 @@ class ARTIQIRGenerator(algorithm.Visitor):
return reduce(lambda l, r: self.append(ir.Arith(ast.Mult(loc=None), l, r)), return reduce(lambda l, r: self.append(ir.Arith(ast.Mult(loc=None), l, r)),
lengths[1:], lengths[0]) lengths[1:], lengths[0])
def _alloate_new_array(self, elt, shape): def _allocate_new_array(self, elt, shape):
total_length = self._get_total_array_len(shape) total_length = self._get_total_array_len(shape)
buffer = self.append(ir.Alloc([total_length], types._TPointer(elt=elt))) buffer = self.append(ir.Alloc([total_length], types._TPointer(elt=elt)))
result_type = builtins.TArray(elt, types.TValue(len(shape.type.elts))) result_type = builtins.TArray(elt, types.TValue(len(shape.type.elts)))
@ -1554,7 +1554,7 @@ class ARTIQIRGenerator(algorithm.Visitor):
# TODO: Broadcasts; select the widest shape. # TODO: Broadcasts; select the widest shape.
# TODO: Detect and special-case matrix multiplication. # TODO: Detect and special-case matrix multiplication.
shape = self.append(ir.GetAttr(lhs, "shape")) shape = self.append(ir.GetAttr(lhs, "shape"))
result = self._alloate_new_array(node.type.find()["elt"], shape) result = self._allocate_new_array(node.type.find()["elt"], shape)
func = self._get_array_binop(node.op, node.type, node.left.type, node.right.type) func = self._get_array_binop(node.op, node.type, node.left.type, node.right.type)
self._invoke_arrayop(func, [result, lhs, rhs]) self._invoke_arrayop(func, [result, lhs, rhs])