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

transforms.inferencer: infer a monomorphic type for slice ":"

This commit is contained in:
whitequark 2016-01-16 03:09:03 +00:00
parent be560dbc63
commit cc22837627
2 changed files with 16 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
artiq/compiler/transforms/inferencer.py
View File

@ -202,12 +202,16 @@ class Inferencer(algorithm.Visitor):
value.loc, None) value.loc, None)
def visit_SliceT(self, node): def visit_SliceT(self, node):
self._unify(node.type, builtins.TInt(), if (node.lower, node.upper, node.step) == (None, None, None):
node.loc, None) self._unify(node.type, builtins.TInt32(),
for operand in (node.lower, node.upper, node.step): node.loc, None)
if operand is not None: else:
self._unify(operand.type, node.type, self._unify(node.type, builtins.TInt(),
operand.loc, None) node.loc, None)
for operand in (node.lower, node.upper, node.step):
if operand is not None:
self._unify(operand.type, node.type,
operand.loc, None)
def visit_ExtSlice(self, node): def visit_ExtSlice(self, node):
diag = diagnostic.Diagnostic("error", diag = diagnostic.Diagnostic("error",

6
lit-test/test/inferencer/slice.py Normal file
View File

@ -0,0 +1,6 @@
# RUN: %python -m artiq.compiler.testbench.inferencer %s >%t
# RUN: OutputCheck %s --file-to-check=%t
x = [0]
# CHECK-L: [::int(width=32)]
x[:] = [1]