gather objects in TPU is not supported

Open
#2,858 4 comments 0 reactions 0 assignees View on GitHub

Maintainers usually reply within 1 day

Nobody has claimed this yet.

  • #4190 by @FinalSunFlower — closed without merging

Assessment

Difficulty
5/5
Estimated time
Over a week
Newbie friendliness
20/100
Issue type
Feature
Clarity
Needs clarification
Activity status
Stale
Tech stack
python

Research direction

Start by reproducing distributed evaluation on a TPU and locating the gather-objects path that raises “gather objects in TPU is not supported.” Check whether TPU object gathering is feasible in this setup; done would require a decided scope, implementation or documented limitation, and validation of the evaluation workflow.

Written by the indexing model from the issue text.

Description

feature request TPU

Do you have any plans to implement the above functionality on a TPU (Also is this even possible on TPU ?) ? I'm trying to distribute evaluation using accelerate.
Error:

    raise NotImplementedError("gather objects in TPU is not supported")
NotImplementedError: gather objects in TPU is not supported

Dominant language
Python
Stars
9.9k
Forks
1.6k
Avg merge
2d 13h
Merged PRs (30d)
13

Getting set up

Open in Codespaces

Starts the project's dev container in your browser, under your own GitHub account.

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

More from huggingface/accelerate

All issues in huggingface/accelerate

Similar issues

More Python issues

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.