Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Appearance settings
This repository was archived by the owner on Jun 5, 2025. It is now read-only.

Modify to use GPU (if available) for embedding#1064

Merged
ptelang merged 1 commit into
mainstacklok/codegate:mainfrom
use-gpu-for-embeddingstacklok/codegate:use-gpu-for-embeddingCopy head branch name to clipboard
Feb 14, 2025
Merged

Modify to use GPU (if available) for embedding#1064
ptelang merged 1 commit into
mainstacklok/codegate:mainfrom
use-gpu-for-embeddingstacklok/codegate:use-gpu-for-embeddingCopy head branch name to clipboard

Conversation

@ptelang

@ptelang ptelang commented Feb 14, 2025

Copy link
Copy Markdown
Contributor

No description provided.

@ptelang ptelang requested a review from JAORMX February 14, 2025 16:22
@lukehinds

Copy link
Copy Markdown

Could I get a little more context Pankaj, is this meaning its using metal or cuda (or both)?

@ptelang ptelang merged commit 1db4075 into main Feb 14, 2025
@ptelang ptelang deleted the use-gpu-for-embedding branch February 14, 2025 19:59
@ptelang

ptelang commented Feb 14, 2025

Copy link
Copy Markdown
Contributor Author

Could I get a little more context Pankaj, is this meaning its using metal or cuda (or both)?

It will use either metal or cuda, whichever is available. The actual implementation of using the GPU is in llama.cpp. GPU was not getting used since codegate was not passing a parameter to the llama.cpp call. This PR fixes the issue by passing that parameter.

@JAORMX

JAORMX commented Feb 14, 2025

Copy link
Copy Markdown
Contributor

Good find @ptelang !

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants

Morty Proxy This is a proxified and sanitized view of the page, visit original site.