Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

when we will support spark -gpu ,now we need spark -gpu feature to train big model #2515

Closed
mullerhai opened this issue Jul 29, 2023 · 3 comments
Labels
enhancement New feature or request stale stale

Comments

@mullerhai
Copy link

mullerhai commented Jul 29, 2023

Hi, I know gluten is perfect tools for spark compute ,now we do some spark gpu work for train big model, need distribute train model with spark rdd/ dataframe data,so if we could support gpu ,maybe it is good

@mullerhai mullerhai added the enhancement New feature or request label Jul 29, 2023
@weiting-chen
Copy link
Contributor

What kind of the GPU work you are asking to support in Gluten?

  1. To run Spark SQL on GPU
  2. To run Spark SQL on CPU, then use Spark RDD/Dataframe to run Train model on GPU directly without writing to the disk.
  3. Other scenarios

@mullerhai
Copy link
Author

What kind of the GPU work you are asking to support in Gluten?

  1. To run Spark SQL on GPU
  2. To run Spark SQL on CPU, then use Spark RDD/Dataframe to run Train model on GPU directly without writing to the disk.
  3. Other scenarios

just 2, as you know ,spark sql compute also complex and need a lot of time ,and train data also need more time ,these are need gpu support , I know your project want to add gpu support ,so ask the project owner

@weiting-chen
Copy link
Contributor

Currently Gluten only focus on pure Spark SQL scenario, for case2, we are working with some customers to integrate ML/DL scenario with Gluten. There will be some works in 2nd half 2023, but no specific time guarantee.

@github-actions github-actions bot added the stale stale label Nov 17, 2023
@github-actions github-actions bot closed this as not planned Won't fix, can't repro, duplicate, stale Nov 25, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request stale stale
Projects
None yet
Development

No branches or pull requests

2 participants