Skip to content

Add hooks for execution on intel gaudi devices - 1#128584

Closed
ankurneog wants to merge 2 commits intopytorch:mainfrom
ankurneog:ankurneog_hpu_updates_phase2
Closed

Add hooks for execution on intel gaudi devices - 1#128584
ankurneog wants to merge 2 commits intopytorch:mainfrom
ankurneog:ankurneog_hpu_updates_phase2

Conversation

@ankurneog
Copy link

Motivation

This is follow up to PR:#126970 to support Gaudi devices for Pytorch UT execution.

Changes

We are adding additional hooks to:

  1. Add dtype exceptions for Gaudi/HPU
  2. Extend onlyNativeDevices decorator functionality to add additional devices

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

Labels

ciflow/trunk Trigger trunk jobs on your pull request Merged open source topic: not user facing topic category triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants