mirror of https://github.com/llvm/torch-mlir
Move numpy compiler support to new directory.
parent
a4f3ce1ed3
commit
1024c508f8
|
@ -0,0 +1,5 @@
|
|||
# Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
||||
# See https://llvm.org/LICENSE.txt for license information.
|
||||
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
||||
|
||||
from .value_coder import *
|
|
@ -8,8 +8,8 @@ from typing import Union
|
|||
|
||||
from _npcomp.mlir import ir
|
||||
|
||||
from . import logging
|
||||
from .interfaces import *
|
||||
from ... import logging
|
||||
from ...interfaces import *
|
||||
|
||||
__all__ = [
|
||||
"CreateNumpyValueCoder",
|
|
@ -11,7 +11,7 @@ from .interfaces import *
|
|||
from .partial_eval_base import *
|
||||
from .target import *
|
||||
from .value_coder_base import *
|
||||
from .value_coder_numpy import *
|
||||
from .extensions import numpy as npc
|
||||
|
||||
|
||||
def create_import_dump_decorator(*,
|
||||
|
@ -33,7 +33,7 @@ def create_import_dump_decorator(*,
|
|||
def create_test_config(target_factory: TargetFactory = GenericTarget64):
|
||||
value_coder = ValueCoderChain([
|
||||
BuiltinsValueCoder(),
|
||||
CreateNumpyValueCoder(),
|
||||
npc.CreateNumpyValueCoder(),
|
||||
])
|
||||
pe_hook = build_default_partial_eval_hook()
|
||||
|
||||
|
|
Loading…
Reference in New Issue