search

tensorflow / tensorflow

An Open Source Machine Learning Framework for Everyone
https://tensorflow.org
Apache License 2.0
182.91k stars 73.92k forks source link

PR #12243: Add Support for FP8 Data Types in cudnn_fusion_compiler. #67149

Closed copybara-service[bot] closed 1 week ago

copybara-service[bot] commented 1 week ago

PR #12243: Add Support for FP8 Data Types in cudnn_fusion_compiler.

Imported from GitHub PR https://github.com/openxla/xla/pull/12243

Copybara import of the project:

-- cee34b537bf34ba3413fa39d01f3267b50eb344a by Elfie Guo elfieg@nvidia.com:

Add support for FP8 datatype in cudnn_fusion_compiler.

Merging this change closes #12243

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/12243 from elfiegg:fp8_triton cee34b537bf34ba3413fa39d01f3267b50eb344a