Yi Zhang 776bd0cccc add lru-cache to avoid repeated calculation

2025-04-04 12:44:26 +08:00

..

include/deep_gemm

Solve STSM bank conflict via padding and 3D TMA

2025-04-03 15:39:35 +08:00

jit

Add DG_NVCC_OVERRIDE_CPP_STANDARD

2025-04-03 15:53:29 +08:00

jit_kernels

add lru-cache to avoid repeated calculation

2025-04-04 12:44:26 +08:00

__init__.py

fix typo

2025-02-26 18:37:22 +08:00

utils.py

Correctly flush L2, as reconstructing the tensors on every iteration effectively put them in the L2, and gave the GPU enough idle time to avoid thermal throttling in a potentially unrealistic way.

2025-03-15 20:46:24 +00:00