fix warning and cache cuda

Signed-off-by: Jacob Nguyen <76754747+jacoobes@users.noreply.github.com>
This commit is contained in:
Jacob Nguyen 2024-08-10 14:03:40 -05:00
parent a616f8da24
commit 3af165f4d9
2 changed files with 10 additions and 1 deletions

View File

@ -1103,11 +1103,20 @@ jobs:
command: |
Invoke-WebRequest -Uri https://sdk.lunarg.com/sdk/download/1.3.261.1/windows/VulkanSDK-1.3.261.1-Installer.exe -OutFile VulkanSDK-1.3.261.1-Installer.exe
.\VulkanSDK-1.3.261.1-Installer.exe --accept-licenses --default-answer --confirm-command install
restore_cache:
keys:
- cudawin118
- run:
name: Install CUDA Toolkit
command: |
Invoke-WebRequest -Uri https://developer.download.nvidia.com/compute/cuda/11.8.0/network_installers/cuda_11.8.0_windows_network.exe -OutFile cuda_11.8.0_windows_network.exe
.\cuda_11.8.0_windows_network.exe -s cudart_11.8 nvcc_11.8 cublas_11.8 cublas_dev_11.8
- save_cache:
key: cudawin118
paths:
- C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.8
- run:
name: Install dependencies
command: |

View File

@ -164,7 +164,7 @@ Napi::Array ChunkedFloatPtr(float *embedding_ptr, int embedding_size, int text_l
// I know there's a way to emplace the raw float ptr into a Napi::Float32Array but idk how and
// im too scared to cause memory issues
// this is goodenough
for (int j = 0; j < chunk.size(); j++)
for (std::vector<float>::size_type j = 0; j < chunk.size(); j++)
{
fltarr.Set(j, chunk[j]);