diff --git a/scripts/unsloth/feature-checks.json b/scripts/unsloth/feature-checks.json index a40de2d49db9..48c4449c755d 100644 --- a/scripts/unsloth/feature-checks.json +++ b/scripts/unsloth/feature-checks.json @@ -129,6 +129,12 @@ { "kind": "backend-op", "op": "GET_ROWS", "params": "type=mxfp8" }, { "kind": "mtmd", "projector": "deepseek41v" } ] + }, + "cuda-fa-quant-small-batch": { + "owner": "unslothai#303", + "checks": [ + { "kind": "backend-op", "op": "FLASH_ATTN_EXT", "params": "n_seq=4" } + ] } }, "unchecked": { diff --git a/scripts/unsloth/pr-set.json b/scripts/unsloth/pr-set.json index bac93cda8858..984dcfde3fca 100644 --- a/scripts/unsloth/pr-set.json +++ b/scripts/unsloth/pr-set.json @@ -40,6 +40,7 @@ "https://github.com/unslothai/llama.cpp/pull/298/commits/d1781c3ea6ab7c75c635a8fea60d2b3f4937c645", "https://github.com/unslothai/llama.cpp/pull/304/commits/64b2a7e7bdddc0c00de67418b4b056cad60666d6", "https://github.com/unslothai/llama.cpp/pull/311/commits/819e95b316743cd2ed7bb54b68f52f5ca1ea820a", - "https://github.com/unslothai/llama.cpp/pull/314/commits/08f367a2a56a7b4405b106eb23a227a910571786" + "https://github.com/unslothai/llama.cpp/pull/314/commits/08f367a2a56a7b4405b106eb23a227a910571786", + "https://github.com/unslothai/llama.cpp/pull/303/commits/34a66f42be7b0624b00e039725ea879fd772b14f" ] }