mmdeploy icon indicating copy to clipboard operation
mmdeploy copied to clipboard

Support for TensorRT 10.0 [Feature]

Open thu-bear opened this issue 1 year ago • 40 comments

Motivation

Thank you very much for your excellent work. MMdeploy has been very helpful to me.

Currently, TensorRT 10.0 has been released, and I found that mmdeploy_tensorrt_ops_obj cannot compile successfully in the TensorRT 10.0 environment.

The current error message is as follows: msvc: [E0144] Type 'const int64_t *' cannot be used to initialize type 'const int *' (gather_topk.cpp line81\82) -> const int * dims = &(inputDesc[0].dims.d[0]);

Related resources

No response

Additional context

No response

thu-bear avatar May 07 '24 02:05 thu-bear

Have you solved this issue yet? What is your mmdeploy version and Visual Studio version?

pingguokiller avatar Jun 01 '24 06:06 pingguokiller

Have you solved this issue yet? What is your mmdeploy version and Visual Studio version?

Not yet:(; tried the latest mmdeploy (V1.3.1), and using VS2022 (V17.1.0)

thu-bear avatar Jun 18 '24 06:06 thu-bear

The breaking changes do not look too bad. I am considering whether to try patching this on a fork or to try to get torch.compile / torch_tensorrt.compile working instead

matthost avatar Jul 25 '24 16:07 matthost

@matthost did you ever end up making the patch on a fork? Any chance you could share it if you did?

jacob-stein avatar Jul 30 '24 23:07 jacob-stein

Not yet, not sure whether I’ll need to after seeing the hardware compatibility feature is as recent as 8.6. Will share if I do. Any reason you want 10?

matthost avatar Jul 31 '24 00:07 matthost

Not a problem, I managed to get it to compile by adding a few casts. No real reason for using V10, other than avoiding future dependency problems.

jacob-stein avatar Jul 31 '24 00:07 jacob-stein

@jacob-stein Would you be able to share the code, if possible?

pictureken avatar Aug 17 '24 06:08 pictureken