2 مستودعات
Applying standard math functions elementwise on tile fragments in GPU kernels.
Distinct from Tile-Based Kernel Authoring: Distinct from Tile-Based Kernel Authoring: focuses on elementwise math operations on tiles, not general tile programming.
Explore 2 awesome GitHub repositories matching artificial intelligence & ml · Elementwise Tile Math Operations. Refine with filters or upvote what's useful.
TileLang is a Python-embedded domain-specific language compiler that JIT-compiles and autotunes GPU kernels. It uses a tile-based DSL, automatic software pipelining, and parallel autotuning to generate optimized GPU kernels at runtime. It supports tensor core operations with Pythonic syntax, automatic memory management, and thread mapping. The compiler searches over tile sizes, thread counts, and scheduling policies, compiling and benchmarking candidates in parallel to find the fastest kernel. It also caches compiled binaries and tuning results to disk for reuse across sessions. TileLang inc
TVM's feature to compute exponential, logarithmic, trigonometric, and other elementary math operations within GPU/CPU kernel code.
xtensor is a C++ multidimensional array library for numerical computing that provides N-dimensional containers with an interface mirroring the NumPy API. It utilizes a lazy evaluation expression engine to defer numerical computations until assignment, which minimizes memory allocations and intermediate copies. The library features a foreign memory array adaptor that allows it to wrap external buffers, such as NumPy arrays, to perform numerical operations in-place without duplicating data. It further optimizes performance through lazy broadcasting and a system that manages the lifetime of temp
Calculates standard elementary math functions, including trigonometric and exponential operations, element-wise.