TensorRT: Myelin error when load ONNX model, Assertion `false && "Invalid size written"' failed

Description

We’re preparing to use Orin with TensorRT8.4 for better performance, but it failed when creating a Builder. Error occurred in myelin which is close-sourced, we’ve gotten stuck here for days and had no clue where to start debugging, so ask for your help:)

Environment

TensorRT Version: 8400 NVIDIA GPU: Orin NVIDIA Driver Version: 11.4 CUDA Version: 11.4 Operating System: Ubunut20.4

Key error

It can be reproduced by just running trtexec with ONNX model, it aborted at error: trtexec: /root/gpgpu/MachineLearning/myelin/src/compiler/optimizer/cublas_impl.cpp:479: void add_heuristic_results_to_tactics(std::vector<cublasLtMatmulHeuristicResult_t>&, std::vectormyelin::ir::tactic_attribute_t&, myelin::ir::tactic_attribute_t&, bool): Assertion `false && “Invalid size written”’ failed.

Detailed log were attached, please check:

test_model.log

Simpilifed onnx model file for reproduction: test_model.tar.gz

About this issue

  • Original URL
  • State: closed
  • Created 2 years ago
  • Comments: 20 (4 by maintainers)

Most upvoted comments

We are working on this, your company also report the same issue using nvonline account, we will update there once have progress, thanks!