mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-05 08:37:06 +08:00
fix typos (#3684)
This commit is contained in:
@@ -49,7 +49,7 @@ class GcuWorker(WorkerBase):
|
||||
def init_device(self):
|
||||
"""Initialize device and Construct model runner"""
|
||||
if paddle.is_compiled_with_custom_device("gcu"):
|
||||
# Set evironment variable
|
||||
# Set environment variable
|
||||
self.device_ids = self.parallel_config.device_ids.split(",")
|
||||
self.device = f"gcu:{self.local_rank}"
|
||||
paddle.device.set_device(self.device)
|
||||
@@ -127,7 +127,7 @@ class GcuWorker(WorkerBase):
|
||||
# NOTE(gongshaotian): may be not need warm_up at this place
|
||||
if self.model_runner.graph_opt_level >= 1:
|
||||
self.model_runner.sot_warmup()
|
||||
# 2. Triger cuda grpah capture
|
||||
# 2. Trigger cuda graph capture
|
||||
self.model_runner.capture_model()
|
||||
set_random_seed(self.fd_config.model_config.seed)
|
||||
|
||||
|
Reference in New Issue
Block a user