mirror of https://github.com/llvm/torch-mlir
fixed some python imports
Change required to enable ./tools/torchscript_e2e_test.sh --config=ireepull/294/head
parent
1c53424fe7
commit
d7320f3bda
|
@ -5,8 +5,8 @@
|
|||
import io
|
||||
import subprocess
|
||||
|
||||
from mlir.ir import *
|
||||
from mlir.passmanager import *
|
||||
from npcomp.ir import *
|
||||
from npcomp.passmanager import *
|
||||
from npcomp.compiler.generic.backend import iree as iree_backend
|
||||
from npcomp.compiler.utils import logging
|
||||
|
||||
|
|
|
@ -4,8 +4,8 @@
|
|||
|
||||
import os
|
||||
|
||||
from mlir.ir import *
|
||||
from mlir.passmanager import *
|
||||
from npcomp.ir import *
|
||||
from npcomp.passmanager import *
|
||||
from npcomp.compiler.generic.backend import refjit as refjit_backend
|
||||
from npcomp.compiler.utils import logging
|
||||
|
||||
|
|
|
@ -6,8 +6,8 @@ import os
|
|||
|
||||
import torch
|
||||
|
||||
from mlir.ir import *
|
||||
from mlir.passmanager import *
|
||||
from npcomp.ir import *
|
||||
from npcomp.passmanager import *
|
||||
from npcomp.compiler.utils import logging
|
||||
|
||||
__all__ = [
|
||||
|
|
|
@ -7,8 +7,8 @@ import os
|
|||
import torch
|
||||
import numpy as np
|
||||
|
||||
from mlir.ir import *
|
||||
from mlir.passmanager import *
|
||||
from npcomp.ir import *
|
||||
from npcomp.passmanager import *
|
||||
from npcomp.compiler.utils import logging
|
||||
import iree.runtime as ireert
|
||||
import iree.compiler as ireec
|
||||
|
|
Loading…
Reference in New Issue