mirror of https://github.com/m-labs/artiq.git
Make division by zero raise an exception.
This commit is contained in:
parent
4cfe4ea148
commit
f8c2709943
|
@ -83,6 +83,10 @@ class TValueError(TException):
|
|||
def __init__(self):
|
||||
super().__init__("ValueError")
|
||||
|
||||
class TZeroDivisionError(TException):
|
||||
def __init__(self):
|
||||
super().__init__("ZeroDivisionError")
|
||||
|
||||
def fn_bool():
|
||||
return types.TConstructor("bool")
|
||||
|
||||
|
@ -107,6 +111,9 @@ def fn_IndexError():
|
|||
def fn_ValueError():
|
||||
return types.TExceptionConstructor("ValueError")
|
||||
|
||||
def fn_ZeroDivisionError():
|
||||
return types.TExceptionConstructor("ZeroDivisionError")
|
||||
|
||||
def fn_range():
|
||||
return types.TBuiltinFunction("range")
|
||||
|
||||
|
|
|
@ -15,6 +15,7 @@ def globals():
|
|||
"Exception": builtins.fn_Exception(),
|
||||
"IndexError": builtins.fn_IndexError(),
|
||||
"ValueError": builtins.fn_ValueError(),
|
||||
"ZeroDivisionError": builtins.fn_ZeroDivisionError(),
|
||||
"len": builtins.fn_len(),
|
||||
"round": builtins.fn_round(),
|
||||
"print": builtins.fn_print(),
|
||||
|
|
|
@ -899,13 +899,16 @@ class ARTIQIRGenerator(algorithm.Visitor):
|
|||
|
||||
def visit_BinOpT(self, node):
|
||||
if builtins.is_numeric(node.type):
|
||||
# TODO: check for division by zero
|
||||
rhs = self.visit(node.right)
|
||||
if isinstance(node.op, (ast.LShift, ast.RShift)):
|
||||
# Check for negative shift amount.
|
||||
self._make_check(self.append(ir.Compare(ast.GtE(loc=None), rhs,
|
||||
ir.Constant(0, rhs.type))),
|
||||
lambda: self.append(ir.Alloc([], builtins.TValueError())))
|
||||
elif isinstance(node.op, (ast.Div, ast.FloorDiv)):
|
||||
self._make_check(self.append(ir.Compare(ast.NotEq(loc=None), rhs,
|
||||
ir.Constant(0, rhs.type))),
|
||||
lambda: self.append(ir.Alloc([], builtins.TZeroDivisionError())))
|
||||
|
||||
return self.append(ir.Arith(node.op, self.visit(node.left), rhs))
|
||||
elif isinstance(node.op, ast.Add): # list + list, tuple + tuple
|
||||
|
|
Loading…
Reference in New Issue