news 2026/3/29 15:47:32

【Triton 教程】triton_language.store

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
【Triton 教程】triton_language.store

Triton 是一种用于并行编程的语言和编译器。它旨在提供一个基于 Python 的编程环境,以高效编写自定义 DNN 计算内核,并能够在现代 GPU 硬件上以最大吞吐量运行。

更多 Triton 中文文档可访问 →triton.hyper.ai/

triton.language.store(pointer, value, mask=None, boundary_check=(), cache_modifier='', eviction_policy='')

将数据张量存储到由指针定义的内存位置。

1.如果pointer是单元素指针,则加载 1 个标量。在这种情况下:

  • mask必须是标量,
  • boundary_checkpadding_option必须为空。

2.如果pointer是 1 个 N 维指针张量,则会存储 1 个 N 维张量。在这种情况下:

  • mask会被隐式地广播到pointer.shape
  • boundary_check必须为空。

3.如果pointer是由make_block_ptr定义的块指针,则会存储 1 个张量。在这种情况下:

  • mask必须为None
  • 可以指定boundary_check以控制越界访问的行为。

value会被隐式地广播为pointer.shape,并转换为pointer.element_ty类型。

参数**:**

  • pointer(triton.PointerType, 或 dtype=triton.PointerType 的块) - 存储 value 元素的内存位置。
  • value(Block) - 要存储的元素张量。
  • masktriton.int1 的块,可选)- 如果 mask[idx] 为 false,则不将 value[idx] 存储在 pointer[idx] 处。
  • boundary_check整数元组,可选)- 整数元组,表示应该进行边界检查的维度。
  • cache_modifier(str,可选,应为 {“”(空字符串)、“.wb”(表示缓存回写所有一致性层级)、“.cg”(表示全局缓存)、“.cs”(表示缓存流)、“.wt”(表示缓存直写)} 中的一个)更多详情请参见缓存操作符)—— 在 NVIDIA PTX 中更改缓存选项。
  • eviction_policystr可选应为 {“”, “evict_first”, “evict_last”} 中的一个)- 更改 NVIDIA PTX 中的驱逐策略。

这个函数也可作为tensor的成员函数调用,使用x.store(...)方式而不是store(x, ...)

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/3/27 4:40:25

5MW风电永磁直驱发电机-1200V直流并网Simulink仿真模型

💥💥💞💞欢迎来到本博客❤️❤️💥💥 🏆博主优势:🌞🌞🌞博客内容尽量做到思维缜密,逻辑清晰,为了方便读者。 ⛳️座右铭&a…

作者头像 李华
网站建设 2026/3/27 6:39:39

学长亲荐10个AI论文软件,自考毕业论文轻松搞定!

学长亲荐10个AI论文软件,自考毕业论文轻松搞定! AI工具如何助你轻松应对论文挑战 在自考学习过程中,毕业论文往往是最让人头疼的一环。从选题到撰写,再到反复修改,每一步都需要投入大量时间和精力。而随着AI技术的不断…

作者头像 李华
网站建设 2026/3/27 17:14:58

中高轨激光通信卫星伺服控制器抗辐照电源模块设计

摘要:随着中高轨道激光通信卫星技术的快速发展,伺服控制器作为激光链路建立与维持的核心执行单元,其电源模块的抗辐照特性成为保障星载设备长期可靠运行的关键要素。本文系统梳理了中高轨空间辐射环境特征及其对电源系统的损伤机理&#xff0…

作者头像 李华