Skip to content

Commit 1928767

Browse files
committed
update ep factory
1 parent 4da9f90 commit 1928767

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

plugin_execution_providers/tensorrt/tensorrt_provider_factory.cc

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -88,17 +88,17 @@ static OrtStatus* ORT_API_CALL CreateEpImpl(OrtEpFactory* this_ptr,
8888
*ep = nullptr;
8989

9090
if (num_devices != 1) {
91-
// we only registered for CPU and only expected to be selected for one CPU
91+
// we only registered for GPU and only expected to be selected for one GPU
9292
// if you register for multiple devices (e.g. CPU, GPU and maybe NPU) you will get an entry for each device
9393
// the EP has been selected for.
9494
return factory->ort_api.CreateStatus(ORT_INVALID_ARGUMENT,
95-
"Example EP only supports selection for one device.");
95+
"TensorRT EP only supports selection for one device.");
9696
}
9797

9898
// Create the execution provider
9999
RETURN_IF_ERROR(factory->ort_api.Logger_LogMessage(logger,
100100
OrtLoggingLevel::ORT_LOGGING_LEVEL_INFO,
101-
"Creating Example EP", ORT_FILE, __LINE__, __FUNCTION__));
101+
"Creating TensorRT EP", ORT_FILE, __LINE__, __FUNCTION__));
102102

103103
// use properties from the device and ep_metadata if needed
104104
// const OrtHardwareDevice* device = devices[0];

0 commit comments

Comments
 (0)