mirror of
https://github.com/ROCm/ROCm.git
synced 2026-02-21 03:00:39 -05:00
053af4e9f8
The purpose of this PR is to remove some circular dependencies and separate concerns better in the frontend. It's still not perfect -- `triton.compile` still includes a few runtime architecture-specific component, but at least much better than before. This PR still assumes that AMD only supports empty kernels right now. Other PRs will follow to make the frontend supports multiple devices in a more modular way.
53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
import ast
|
|
from typing import Optional, Union
|
|
|
|
|
|
class CompilationError(Exception):
|
|
source_line_count_max_in_message = 12
|
|
|
|
def _format_message(self) -> str:
|
|
node = self.node
|
|
if self.src is None:
|
|
source_excerpt = " <source unavailable>"
|
|
else:
|
|
source_excerpt = self.src.split('\n')[:node.lineno][-self.source_line_count_max_in_message:]
|
|
if source_excerpt:
|
|
source_excerpt.append(' ' * node.col_offset + '^')
|
|
source_excerpt = '\n'.join(source_excerpt)
|
|
else:
|
|
source_excerpt = " <source empty>"
|
|
|
|
message = "at {}:{}:{}".format(node.lineno, node.col_offset, source_excerpt)
|
|
if self.error_message:
|
|
message += '\n' + self.error_message
|
|
return message
|
|
|
|
def __init__(self, src: Optional[str], node: ast.AST, error_message: Union[str, None]):
|
|
self.src = src
|
|
self.node = node
|
|
self.error_message = error_message
|
|
self.message = self._format_message()
|
|
|
|
def set_source_code(self, src: Optional[str]):
|
|
self.src = src
|
|
self.message = self._format_message()
|
|
|
|
def __str__(self):
|
|
return self.message
|
|
|
|
def __repr__(self):
|
|
return "{}({!r})".format(type(self).__name__, self.message)
|
|
|
|
def __reduce__(self):
|
|
# this is necessary to make CompilationError picklable
|
|
return type(self), (self.src, self.node, self.error_message)
|
|
|
|
|
|
class CompileTimeAssertionFailure(CompilationError):
|
|
"""Specific exception for failed tests in `static_assert` invocations"""
|
|
pass
|
|
|
|
|
|
class UnsupportedLanguageConstruct(CompilationError):
|
|
pass
|