小红花·文摘
  • 首页
  • 广场
  • 排行榜🏆
  • 直播
  • FAQ
Dify.AI

NVIDIA的Tensor Core专为加速矩阵运算而设计,TN布局在GEMM问题中表现最佳,优化了内存访问和缓存利用率,从而显著提升性能。大多数NVIDIA GPU架构仅支持TN布局的MMA指令。

NVIDIA Tensor Core TN布局MMA指令

Lei Mao's Log Book
Lei Mao's Log Book · 2025-12-06T08:00:00Z

NVIDIA GPU的峰值AI性能通常以TFLOPS或TOPS表示,但使用HPC软件重现这些性能并不总是有效。通过自定义微基准测试直接调用Tensor Core MMA指令,可以更可靠地测量性能。本文介绍了如何使用CUTLASS和CuTe测量NVIDIA Tensor Core MMA指令的峰值性能,并提供选择合适指令的参考。

基准测试NVIDIA Tensor Core MMA指令的峰值性能

Lei Mao's Log Book
Lei Mao's Log Book · 2025-11-26T08:00:00Z

MMA(矩阵乘法和累加)是GEMM的核心操作。CuTe提供API以配置MMA原子和瓦片,支持更大规模的问题解决。本文讨论了CuTe的瓦片MMA配置、布局及API示例,展示了如何优化性能,通过静态共享内存和线程划分来实现。

CuTe 瓦片 MMA

Lei Mao's Log Book
Lei Mao's Log Book · 2025-01-09T08:00:00Z

近年来,文本到图像(T2I)模型取得了显著进展并广泛应用。然而,这一进展也带来了滥用的潜在风险。研究人员引入了MMA-Diffusion框架,该框架能够绕过当前的防御措施,揭示现有防御机制的弱点。该框架利用文本和视觉模式来绕过提示过滤器和事后安全检查器等保护措施。

MMA-Diffusion:多模态对抗扩散模型

BriefGPT - AI 论文速递
BriefGPT - AI 论文速递 · 2023-11-29T00:00:00Z

MAMBA是一种基于深度学习的平台,可以处理多种成像模态的3D组织图像并预测患者结果。该框架使用3D基于块的方法,在预测方面表现出比传统的基于单层切片的预测方法更好的性能。MAMBA为临床决策支持提供了一种通用有效的3D弱监督学习框架,可以帮助揭示新的3D形态生物标志物以进行预后和治疗反应的研究。

MMA-Net:自动柯布角度测量的多形态感知网络

BriefGPT - AI 论文速递
BriefGPT - AI 论文速递 · 2023-09-25T00:00:00Z
  • <<
  • <
  • 1 (current)
  • >
  • >>
👤 个人中心
在公众号发送验证码完成验证
登录验证
在本设备完成一次验证即可继续使用

完成下面两步后,将自动完成登录并继续当前操作。

1 关注公众号
小红花技术领袖公众号二维码
小红花技术领袖
如果当前 App 无法识别二维码,请在微信搜索并关注该公众号
2 发送验证码
在公众号对话中发送下面 4 位验证码
友情链接: MOGE.AI 九胧科技 模力方舟 Gitee AI 菜鸟教程 Remio.AI DeekSeek连连 53AI 神龙海外代理IP IPIPGO全球代理IP 东波哥的博客 匡优考试在线考试系统 开源服务指南 蓝莺IM Solo 独立开发者社区 AI酷站导航 极客Fun 我爱水煮鱼 周报生成器 He3.app 简单简历 白鲸出海 T沙龙 职友集 TechParty 蟒周刊 Best AI Music Generator

小红花技术领袖俱乐部
小红花·文摘:汇聚分发优质内容
小红花技术领袖俱乐部
Copyright © 2021-
粤ICP备2022094092号-1
公众号 小红花技术领袖俱乐部公众号二维码
视频号 小红花技术领袖俱乐部视频号二维码