mirror of https://github.com/llvm/torch-mlir
90 lines
2.2 KiB
Python
90 lines
2.2 KiB
Python
|
# 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 typing import *
|
||
|
|
from _npcomp.mlir import ir
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
"GenericTarget32",
|
||
|
|
"GenericTarget64",
|
||
|
|
"Target",
|
||
|
|
"TargetFactory",
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
class Target:
|
||
|
|
"""
|
||
|
|
Abstract class providing configuration and hooks for a specific compilation
|
||
|
|
target.
|
||
|
|
"""
|
||
|
|
__slots__ = [
|
||
|
|
"_mlir_helper",
|
||
|
|
]
|
||
|
|
|
||
|
|
def __init__(self, mlir_helper: ir.DialectHelper):
|
||
|
|
super().__init__()
|
||
|
|
self._mlir_helper = mlir_helper
|
||
|
|
|
||
|
|
@property
|
||
|
|
def mlir_helper(self):
|
||
|
|
return self._mlir_helper
|
||
|
|
|
||
|
|
@property
|
||
|
|
def mlir_context(self):
|
||
|
|
return self._mlir_helper.context
|
||
|
|
|
||
|
|
@property
|
||
|
|
def target_name(self) -> str:
|
||
|
|
return NotImplementedError()
|
||
|
|
|
||
|
|
@property
|
||
|
|
def impl_int_type(self) -> ir.Type:
|
||
|
|
"""Gets the default int type for the backend for the Python 'int' type."""
|
||
|
|
raise NotImplementedError()
|
||
|
|
|
||
|
|
@property
|
||
|
|
def impl_float_type(self) -> ir.Type:
|
||
|
|
"""Gets the implementation's type for the python 'float' type."""
|
||
|
|
raise NotImplementedError()
|
||
|
|
|
||
|
|
|
||
|
|
class GenericTarget64(Target):
|
||
|
|
"""A generic 64 bit target."""
|
||
|
|
|
||
|
|
@property
|
||
|
|
def target_name(self) -> str:
|
||
|
|
return "generic64"
|
||
|
|
|
||
|
|
@property
|
||
|
|
def impl_int_type(self) -> ir.Type:
|
||
|
|
"""Gets the default int type for the backend for the Python 'int' type."""
|
||
|
|
return self.mlir_helper.i64_type
|
||
|
|
|
||
|
|
@property
|
||
|
|
def impl_float_type(self) -> ir.Type:
|
||
|
|
"""Gets the implementation's type for the python 'float' type."""
|
||
|
|
return self.mlir_helper.f64_type
|
||
|
|
|
||
|
|
|
||
|
|
class GenericTarget32(Target):
|
||
|
|
"""A generic 32 bit target (uses 32bit ints and floats)."""
|
||
|
|
|
||
|
|
@property
|
||
|
|
def target_name(self) -> str:
|
||
|
|
return "generic32"
|
||
|
|
|
||
|
|
@property
|
||
|
|
def impl_int_type(self) -> ir.Type:
|
||
|
|
"""Gets the default int type for the backend for the Python 'int' type."""
|
||
|
|
return self.mlir_helper.i32_type
|
||
|
|
|
||
|
|
@property
|
||
|
|
def impl_float_type(self) -> ir.Type:
|
||
|
|
"""Gets the implementation's type for the python 'float' type."""
|
||
|
|
return self.mlir_helper.f32_type
|
||
|
|
|
||
|
|
|
||
|
|
# Factory for producing a target (matches the Target constructor).
|
||
|
|
TargetFactory = Callable[[ir.DialectHelper], Target]
|