Commit a83e1e1e by Jian Weng Committed by Tianqi Chen

Import hybrid module in __init__.py (#1323)

parent db4be63c
...@@ -16,6 +16,7 @@ from . import node ...@@ -16,6 +16,7 @@ from . import node
from . import ir_builder from . import ir_builder
from . import target from . import target
from . import generic from . import generic
from . import hybrid
from . import ndarray as nd from . import ndarray as nd
from .ndarray import context, cpu, gpu, opencl, cl, vulkan, metal, mtl from .ndarray import context, cpu, gpu, opencl, cl, vulkan, metal, mtl
......
...@@ -35,7 +35,7 @@ class HybridParser(ast.NodeVisitor): ...@@ -35,7 +35,7 @@ class HybridParser(ast.NodeVisitor):
ast.Add : operator.add, ast.Add : operator.add,
ast.Sub : operator.sub, ast.Sub : operator.sub,
ast.Mult : operator.mul, ast.Mult : operator.mul,
ast.Div : _make.Div, ast.Div : operator.div if sys.version_info[0] == 2 else operator.truediv,
ast.Mod : operator.mod, ast.Mod : operator.mod,
ast.BitOr : operator.or_, ast.BitOr : operator.or_,
ast.BitAnd: operator.and_, ast.BitAnd: operator.and_,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment