torch-mlir/python/npcomp/compiler/target.py

90 lines
2.2 KiB
Python
Raw Normal View History

# 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]