chore: Upgrade TensorRT version to TRT 10 EA by peri044 · Pull Request #2699 · pytorch/TensorRT (original) (raw)

build_serialized_network returns a plan instead of engine, which is different from previous TRT. In TRT-10, we can do two steps to get an engine:

plan = builder.build_serialized_network(network, config)
engine = runtime.deserialize_cuda_engine(plan)

If it is on purpose to get plan, maybe we can change the name to make it easy understand.