diff --git a/bitsandbytes/triton/matmul_perf_model.py b/bitsandbytes/triton/matmul_perf_model.py index 2080df10f..199ceb1a3 100644 --- a/bitsandbytes/triton/matmul_perf_model.py +++ b/bitsandbytes/triton/matmul_perf_model.py @@ -1,3 +1,6 @@ +# Adapted from https://github.com/triton-lang/kernels/blob/eeeebdd8be7d13629de22d600621e6234057eed3/kernels/matmul_perf_model.py +# https://github.com/triton-lang/kernels is licensed under the MIT License. + import functools import heapq