openmlsys/openmlsys-cuda

This is stars and forks stats for /openmlsys/openmlsys-cuda repository. As of 10 May, 2024 this repository has 77 stars and 8 forks.

openmlsys-cuda Examples for beginners to write your own high-performance AI operators. We introduced optimizations tricks like using shared memory and pipeline rearrangement to maximize the throughput. We also provided an example for using CUTLASS to implement an FC + ReLU fused operator. Dependencies Eigen: CPU linear algebra template library OpenMP: Enable multi-threads acceleration on CPU CUDA toolkit: Compile GPU kernels and analyse GPU executions Gflags: Commandline flags library released by...