o }oiQ@sddlZddlZddlmZddlZddlZddlmZedd\Z Z edd\Z Z dej d efd d Zdej d efd d Zejddd eedffddZdS)N)Tuple)safe_import_fromz(transformer_engine.pytorch.float8_tensor Float8Tensorz'transformer_engine.pytorch.mxfp8_tensor MXFP8TensortensorreturncCtot|tS)z6Check if a tensor is a Transformer Engine Float8Tensor)HAVE_TE_FLOAT8TENSOR isinstancerrr G/home/ubuntu/.local/lib/python3.10/site-packages/nemo/utils/te_utils.pyis_float8tensorrcCr)z5Check if a tensor is a Transformer Engine MXFP8Tensor)HAVE_TE_MXFP8TENSORr MXFP8r r r r is_mxfp8tensor!rr)maxsize.cCstjtjdjS)zTransformer Engine versionztransformer-engine) packagingversionVersion importlibmetadatareleaser r r r te_version&sr) functoolsimportlib.metadatartypingrrtorchnemo.utils.import_utilsrrr rrTensorboolrr lru_cacheintrr r r r s