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

Use torch_tensorrt.Device instead of torch.device in trt compile #8051

Merged
merged 2 commits into from
Aug 29, 2024

Conversation

KumoLiu
Copy link
Contributor

@KumoLiu KumoLiu commented Aug 28, 2024

Fixes #8050

Description

A few sentences describing the changes proposed in this pull request.

Types of changes

  • Non-breaking change (fix or new feature that would not break existing functionality).
  • Breaking change (fix or new feature that would cause existing functionality to change).
  • New tests added to cover the changes.
  • Integration tests passed locally by running ./runtests.sh -f -u --net --coverage.
  • Quick tests passed locally by running ./runtests.sh --quick --unittests --disttests.
  • In-line docstrings updated.
  • Documentation updated, tested make html command in the docs/ folder.

@KumoLiu
Copy link
Contributor Author

KumoLiu commented Aug 29, 2024

/build

@KumoLiu KumoLiu enabled auto-merge (squash) August 29, 2024 02:58
@KumoLiu KumoLiu merged commit 29ce1a7 into Project-MONAI:dev Aug 29, 2024
28 checks passed
@KumoLiu KumoLiu deleted the trt branch August 29, 2024 05:51
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

AttributeError: 'torch.device' object has no attribute 'gpu_id'
2 participants