[Bucketing] Prompt with 0 min and max context blocks (#534)

Signed-off-by: Agata Dobrzyniewicz <adobrzyniewicz@habana.ai>
Signed-off-by: Agata Dobrzyniewicz <160237065+adobrzyn@users.noreply.github.com>
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Author: adobrzyn

vllm_gaudi/extension/bucketing/linear.py

@@ -132,6 +132,8 @@ def warmup_range(config: Tuple[int, int, int]):
                           "batch size. If you want to skip warmup, "
                           "set VLLM_SKIP_WARMUP=true")
     if add_zero_bucket:
+        if bmin == 0 and bmax == 0:
+            return [0]
         bmin = bstep
     base = itertools.repeat(2)
     ramp_up_acc = itertools.accumulate(base, func=operator.mul, initial=bmin)