摩尔线程开源高性能线性代数模板库 MUTLASS,加速基于国产 GPU 的算子开发以及算法创新

摩尔线程开源高性能线性代数模板库 MUTLASS,加速基于国产 GPU 的算子开发以及算法创新
感谢霍州市融媒体中心信息网网网友 Thorn_Flower 的线索投递!

霍州市融媒体中心信息网网 11 月 12 日消息,摩尔线程宣布开源高性能线性代数模板库 MUTLASS,以便开发者能够更高效针对摩尔线程全功能 GPU 的 MUSA Core 及 Tensor Core 等单元进行编程,加速基于国产 GPU 的算子开发以及算法创新。

作为摩尔线程专为 MUSA 架构优化的高性能计算库,MUTLASS(MUSA Templates for Linear Algebra Subroutines)是基于开源模板库 CUTLASS 进行的 MUSA 适配和定制化开发、优化

针对矩阵乘法及相关变种,MUTLASS 提供了一系列高性能的 C++ 模板组件,并采用了与 muDNN 库类似的分层分解及数据搬运策略,以确保性能的充分发挥。

在本次开源的版本中,摩尔线程适配了 CuTe 后端库,为其增加了第三代 MUSA 架构的 MMA 计算原语,支持 TF32 / FP16 / BF16 / INT8 等多种数据精度,并以此为基础,初步实现了矩阵乘法、默认实例库、性能测试器及相关工具包的支持。

霍州市融媒体中心信息网网附 MUTLASS 开源地址:https://github.com/MooreThreads/mutlass

特别声明:[摩尔线程开源高性能线性代数模板库 MUTLASS,加速基于国产 GPU 的算子开发以及算法创新] 该文观点仅代表作者本人,今日霍州系信息发布平台,霍州网仅提供信息存储空间服务。

猜你喜欢

《完美世界》:秦怡宁与石昊的羁绊及离散之殇(《完美世界》TXT下载全本)

在生死攸关的关头,秦怡宁与石子陵选择了妥协,他们的第二个儿子秦昊因此诞生,但石昊的命运依旧未能改观,秦怡宁在不老山的囚禁中日日以泪洗面,心中满是对石昊的牵挂和担忧。此时,已经成长为荒天帝的石昊,为了保护父母不…

《完美世界》:秦怡宁与石昊的羁绊及离散之殇(《完美世界》TXT下载全本)

光子计数(Photon Counting)(光子计数ct与普通CT的区别)

光子计数是低光水平下的光电探测技术,它的诞生源于对极限光电探测的需求。在许多科学研究和工程应用中,如量子信息处理、生物医学成像、天文观测等,都需要在极低光水平下进行光电探测。在天文观测中,光子计数可以用于实…

光子计数(Photon Counting)(光子计数ct与普通CT的区别)

第10号台风白鹿生成(台风白鹿最新路径)

台风生成!今年第10号台风“白鹿”(热带风暴级;英文名称:Bailu;名字来源:中国;名称意义:白色的鹿,中文意指吉祥)已于今天上午在日本以南洋面生成。 预计,“白鹿”将以每小时20-25公里的速度向东北方向…

第10号台风白鹿生成(台风白鹿最新路径)

比比皆是的比比是啥?(比比皆是的含义是什么?)

多得很,到处都是。 也存在多种的说法。 就单看今天的比, 比字的常见用法, 是这几个关键词: 比比,一共两个比, 有遍布各处的意味。比比皆是,到处都是。 一个挨一个地相邻排列, 形容房屋等密集的样子。 比…

比比皆是的比比是啥?(比比皆是的含义是什么?)

PCB板清洗干燥机配套高压鼓风机2HB810-7AH27(pcb板清洗设备)

高效干燥与切水风刀设计:高压气流通过风刀形成均匀气幕,快速吹除PCB板表面残留水分,缩短干燥时间,提升生产效率。其他电子设备制造液晶显示器(LCD)、电脑显示器(TFT)生产:清洗与干燥工艺同PCB板类似,…

PCB板清洗干燥机配套高压鼓风机2HB810-7AH27(pcb板清洗设备)