feat(//py/torch_tensorrt/dynamo): Support for BF16 by narendasan · Pull Request #2833 · pytorch/TensorRT (original) (raw)

tensorrt module doesn't have bf16 attribute.

Python 3.10.12 (main, Nov 20 2023, 15:14:05) [GCC 11.4.0] on linux Type "help", "copyright", "credits" or "license" for more information.

import tensorrt tensorrt.version '10.0.1' hasattr(tensorrt, "bf16") False hasattr(tensorrt, "bfloat16") True