尧图网站建设 尧图网络
  • 首页
  • 关于我们
  • 服务项目
  • 案例展示
  • 建站流程
  • 资讯中心
  • 联系我们
首页/资讯中心/详情

【Triton 教程】triton_language.store

【Triton 教程】triton_language.store
📅 发布时间:2026/6/20 0:59:01

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

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

相关新闻

  • linux上面能对pdf注释嘛?推荐Okular,亲测好用
  • 机器人自主学习新技巧:强化学习的革命
  • LuatOS平台下USB系统可靠性设计:硬件基础与开发进阶!

最新新闻

  • MPC5604P外部中断与DSPI时序参数深度解析与工程实践
  • DFT仿真实战:从STUCK-AT到AT-SPEED的验证要点解析
  • ReadCat安全最佳实践:终极插件安全与用户数据保护指南
  • 2026 上海权威数据 + 真实用户口碑|靠谱空调维修首选上海迪迅通制冷设备 - 星际AI
  • 从零开始:PaddleX如何让AI开发像搭积木一样简单?
  • 抖店无货源铺货怎么不违规?拼多多商品违规检测新手合规教程 - 抖掌柜

日新闻

  • 信任的进化:技术实现详解——如何用JavaScript构建博弈论模拟器
  • Terrakube自定义工作流:如何集成OPA、Infracost等工具扩展IaC能力
  • grunt-concurrent快速入门:5分钟学会并行运行Grunt任务

周新闻

  • 3步解锁iOS设备:applera1n激活锁绕过完全指南
  • 39 2026 人工智能证书终极盘点,普通人选 AI 证书可以从这些方向入手
  • Redis 暴露公网有多危险?从端口检查到补救步骤

月新闻

  • 【总结】入门篇:50句话让你记住架构核心概念
  • WeChatMsg技术方案解析:实现Mac微信数据自主管理的完整解决方案
  • WeChatMsg:革新性微信数据备份方案,打造你的专属数字记忆库

关于尧图

  • 公司简介
  • 团队介绍
  • 企业文化
  • 荣誉资质

服务项目

  • 定制开发
  • 电商建站
  • UI 设计
  • 运维服务

快速链接

  • 案例展示
  • 建站流程
  • 常见问题
  • 资讯中心

联系方式

  • 📍北京市朝阳区互联网产业园 A 座 10 层
  • 📞400-888-8888
  • ✉️contact@rkmt.cn
  • 🕐周一至周日 9:00-21:00

© 2024 北京尧图网络科技有限公司 版权所有 | 京 ICP 备 XXXXXXXX 号