-
Notifications
You must be signed in to change notification settings - Fork 370
Llama distributed example #3831
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
Open
apbose
wants to merge
3
commits into
abose/trt_llm_installation_dist
Choose a base branch
from
abose/torchTRT_dist_llama
base: abose/trt_llm_installation_dist
Could not load branches
Branch not found: {{ refName }}
Loading
Could not load tags
Nothing to show
Loading
Are you sure you want to change the base?
Some commits from the old base branch may be removed from the timeline,
and old review comments may become outdated.
Open
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
There are some changes that do not conform to Python style guidelines:
--- /home/runner/work/TensorRT/TensorRT/py/torch_tensorrt/dynamo/utils.py 2025-09-22 06:34:48.499754+00:00
+++ /home/runner/work/TensorRT/TensorRT/py/torch_tensorrt/dynamo/utils.py 2025-09-22 06:35:25.106098+00:00
@@ -863,6 +863,6 @@
return False
def is_thor() -> bool:
if torch.cuda.get_device_capability() in [(11, 0)]:
- return True
\ No newline at end of file
+ return True
6e99bbc
to
7134053
Compare
68dcaef
to
ca70a9d
Compare
ca70a9d
to
5700249
Compare
5700249
to
b59abbf
Compare
3f1fa7e
to
54948d9
Compare
b59abbf
to
480fb0a
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
There are some changes that do not conform to Python style guidelines:
--- /home/runner/work/TensorRT/TensorRT/py/torch_tensorrt/dynamo/distributed/utils.py 2025-09-25 20:01:17.061900+00:00
+++ /home/runner/work/TensorRT/TensorRT/py/torch_tensorrt/dynamo/distributed/utils.py 2025-09-25 20:01:51.633567+00:00
@@ -100,11 +100,10 @@
return True
except Exception as e:
logger.warning(f"Failed to detect CUDA version: {e}")
return False
-
return True
def _cache_root() -> Path:
2bbc423
to
5beefc0
Compare
5beefc0
to
809c7ee
Compare
480fb0a
to
f00d349
Compare
bb6ec7a
to
dd88017
Compare
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Labels
cla signed
component: api [Python]
Issues re: Python API
component: dynamo
Issues relating to the `torch.compile` or `torch._dynamo.export` paths
component: lowering
Issues re: The lowering / preprocessing passes
component: runtime
component: torch_compile
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
The PR addresses