From 7394748dea7b6482f8eb497bff70cd83cb75ee3a Mon Sep 17 00:00:00 2001
From: Ben Mares <services-git-throwaway1@tensorial.com>
Date: Sat, 25 Jan 2025 11:02:45 -0800
Subject: [PATCH] Try adding a bunch of junk from pytorch recipe

---
 recipes/vllm/recipe.yaml | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/recipes/vllm/recipe.yaml b/recipes/vllm/recipe.yaml
index 54d32e1de9267..39f5b56013c80 100644
--- a/recipes/vllm/recipe.yaml
+++ b/recipes/vllm/recipe.yaml
@@ -35,6 +35,23 @@ requirements:
   - ${{ compiler('c') }}
   - ${{ compiler('cxx') }}
   - ${{ compiler('cuda') }}
+  - cuda-version {{ cuda_compiler_version }}
+  - nccl
+  - cuda-driver-dev                 # [linux]
+  - cuda-cudart-dev
+  - cuda-cupti-dev    # [linux]
+  - cuda-nvrtc-dev
+  - cuda-nvtx-dev
+  - cuda-nvml-dev
+  - cuda-profiler-api
+  - cusparselt     # [linux]
+  - libcublas-dev
+  - libcudss-dev   # [linux]
+  - libcufile-dev  # [linux]
+  - libcufft-dev
+  - libcurand-dev
+  - libcusolver-dev
+  - libcusparse-dev
   run:
   - python
   - psutil