mirror of
https://github.com/exaloop/codon.git
synced 2025-06-03 15:03:52 +08:00
e1016f9e9a
* Upgrade to LLVM 15 (WIP) * Call `setPresplitCoroutine()` on coro LLVM funcs * Use new pass manager * Update deps * Update docs * Fix exceptions on M1 * Add libunwind * Use Orc and JITLink for "codon run" * JITLink integration * Fix callback * Fix strbuf, fix GC root registration * Fix test init * Fix llvm function * Fix pickle, float atomics * Add TargetLibraryAnalysis * Use new LLVM pass manager in GPU codegen * Fix libunwind linking * Fix libunwind * Fix GPU passes * Don't link libunwind explicitly * Bump version * Update plugins for new pass manager * Fix bash error * Fix GPU GV extraction * Move simd module to experimental folder * Update file read * Add benchmark programs * Add dynamic tuple * Fix parser tuple slicing bug * Make DynamicTuple interoperable with Tuple * Fix DynamicTuple GPU interop * Dockerize builds * Simplify CMake * Revert "Simplify CMake" This reverts commit 08d2920349b5033750b54b4fb7aaa9bac264b750. Co-authored-by: Ibrahim Numanagić <ibrahimpasa@gmail.com>
140 lines
3.4 KiB
Python
140 lines
3.4 KiB
Python
# (c) 2022 Exaloop Inc. All rights reserved.
|
|
|
|
# Core library
|
|
|
|
from internal.core import *
|
|
from internal.attributes import *
|
|
from internal.types.ptr import *
|
|
from internal.types.str import *
|
|
from internal.types.int import *
|
|
from internal.types.bool import *
|
|
from internal.types.array import *
|
|
from internal.types.error import *
|
|
from internal.types.intn import *
|
|
from internal.types.float import *
|
|
from internal.types.byte import *
|
|
from internal.types.generator import *
|
|
from internal.types.optional import *
|
|
from internal.types.slice import *
|
|
from internal.types.range import *
|
|
from internal.internal import *
|
|
from internal.types.strbuf import strbuf as _strbuf
|
|
from internal.types.collections.list import *
|
|
import internal.c_stubs as _C
|
|
|
|
|
|
def next(g: Generator[T], default: Optional[T] = None, T: type) -> T:
|
|
if g.done():
|
|
if default:
|
|
return unwrap(default)
|
|
else:
|
|
raise StopIteration()
|
|
return g.next()
|
|
|
|
|
|
from C import seq_print_full(str, cobj)
|
|
|
|
|
|
class Set:
|
|
items: List[T]
|
|
T: type
|
|
|
|
def __init__(self):
|
|
self.items = []
|
|
|
|
def __iter__(self) -> Generator[T]:
|
|
yield from self.items
|
|
|
|
def add(self, what: T):
|
|
if what not in self.items:
|
|
self.items.append(what)
|
|
|
|
def __repr__(self) -> str:
|
|
s = self.items.__repr__()
|
|
s.ptr[0] = "{".ptr[0]
|
|
s.ptr[s.len - 1] = "}".ptr[0]
|
|
return s
|
|
|
|
|
|
class Dict:
|
|
keys: List[K]
|
|
values: List[V]
|
|
K: type
|
|
V: type
|
|
|
|
def __init__(self):
|
|
self.keys = []
|
|
self.values = []
|
|
|
|
def __iter__(self) -> Generator[K]:
|
|
yield from self.keys
|
|
|
|
def items(self) -> Generator[Tuple[K, V]]:
|
|
for i in range(self.keys.len):
|
|
yield (self.keys[i], self.values[i])
|
|
|
|
def __contains__(self, key: K) -> bool:
|
|
return self.keys.index(key) != -1
|
|
|
|
def __getitem__(self, key: K) -> V:
|
|
i = self.keys.index(key)
|
|
return self.values[i]
|
|
|
|
def __setitem__(self, key: K, val: V):
|
|
i = self.keys.index(key)
|
|
if i != -1:
|
|
self.values[i] = val
|
|
else:
|
|
self.keys.append(key)
|
|
self.values.append(val)
|
|
|
|
def __len__(self) -> int:
|
|
return self.keys.len
|
|
|
|
def __repr__(self) -> str:
|
|
n = self.__len__()
|
|
if n == 0:
|
|
return "{}"
|
|
else:
|
|
lst = []
|
|
lst.append("{")
|
|
first = True
|
|
for k, v in self.items():
|
|
if not first:
|
|
lst.append(", ")
|
|
else:
|
|
first = False
|
|
lst.append(k.__repr__())
|
|
lst.append(": ")
|
|
lst.append(v.__repr__())
|
|
lst.append("}")
|
|
return str.cat(lst)
|
|
|
|
|
|
@extend
|
|
class str:
|
|
def __getitem__(self, idx: int) -> str:
|
|
if idx < 0:
|
|
idx += self.len
|
|
if not (0 <= idx < self.len):
|
|
raise IndexError("string index out of range")
|
|
return str(self.ptr + idx, 1)
|
|
|
|
def __getitem__(self, s: Slice) -> str:
|
|
if s.start is None and s.stop is None and s.step is None:
|
|
return self.__copy__()
|
|
elif s.step is None:
|
|
start, stop, step, length = s.adjust_indices(self.len)
|
|
return str(self.ptr + start, length)
|
|
else:
|
|
raise ValueError("nope")
|
|
|
|
def __repr__(self) -> str:
|
|
return f"'{self}'"
|
|
|
|
|
|
from internal.builtin import *
|
|
|
|
from openmp import Ident as __OMPIdent, for_par
|
|
from internal.dlopen import dlsym as _dlsym
|