Skip to content

Commit 402d703

Browse files
committed
minor change to function name, with change in base branch
1 parent 20f864e commit 402d703

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/distributed_inference/tensor_parallel_llama3.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,11 +24,11 @@
2424
import torch_tensorrt
2525
from torch_tensorrt.dynamo.distributed.utils import (
2626
get_tensor_parallel_device_mesh,
27-
initialize_logger,
27+
initialize_distributed_logger,
2828
)
2929

3030
device_mesh, _world_size, _rank = get_tensor_parallel_device_mesh()
31-
logger = initialize_logger(_rank, "tensor_parallel_llama3")
31+
logger = initialize_distributed_logger(_rank, "tensor_parallel_llama3")
3232

3333
logger.info(f"Starting PyTorch TP example on rank {_rank}.")
3434
assert (

0 commit comments

Comments
 (0)