Live
Black Hat USADark ReadingBlack Hat AsiaAI BusinessTech creators are getting the star treatment at a new talent management firmBusiness InsiderBaidu’s robotaxis froze in traffic creating chaosThe Verge AI9 companies that have done AI-related layoffsBusiness InsiderSlack's upgraded AI can analyze how you workEngadgetTown hall in Bay Ridge spotlights AI concerns in NYC public schools - BKReaderGoogle News: AI SafetyOpenAI announces new ‘human powered’ ChatGPT-6 - huckmag.comGoogle News: ChatGPTGoogle Fixes AI Coding Agents' Outdated Code Problem - The Tech BuzzGoogle News: DeepMindThese car gadgets are worth every pennyZDNet AIGoogle Faces Demands to Prohibit AI Videos for Kids on YouTubeBloomberg TechnologyContributor: Investigate the AI campaigns flooding public agencies with fake comments - Los Angeles TimesGoogle News: AIWhy Enterprise AI Stalls Before It Scales - AI BusinessGoogle News: Generative AIThese pocket-sized tech gadgets are packed with purpose (and they're inexpensive)ZDNet AIBlack Hat USADark ReadingBlack Hat AsiaAI BusinessTech creators are getting the star treatment at a new talent management firmBusiness InsiderBaidu’s robotaxis froze in traffic creating chaosThe Verge AI9 companies that have done AI-related layoffsBusiness InsiderSlack's upgraded AI can analyze how you workEngadgetTown hall in Bay Ridge spotlights AI concerns in NYC public schools - BKReaderGoogle News: AI SafetyOpenAI announces new ‘human powered’ ChatGPT-6 - huckmag.comGoogle News: ChatGPTGoogle Fixes AI Coding Agents' Outdated Code Problem - The Tech BuzzGoogle News: DeepMindThese car gadgets are worth every pennyZDNet AIGoogle Faces Demands to Prohibit AI Videos for Kids on YouTubeBloomberg TechnologyContributor: Investigate the AI campaigns flooding public agencies with fake comments - Los Angeles TimesGoogle News: AIWhy Enterprise AI Stalls Before It Scales - AI BusinessGoogle News: Generative AIThese pocket-sized tech gadgets are packed with purpose (and they're inexpensive)ZDNet AI

ciflow/torchtitan/178947: Update on "add API to check if a tensor is symm-mem-tensor"

PyTorch Releasesby shunting314April 1, 20261 min read0 views
Source Quiz

<p>In Helion autotuner, we need clone a input symm memory tensor properly if the kernel inplace update it. That requires us to know if a tensor is a symm memory tensor. Right now I call rendezvous for the tensor. If no exception is thrown, then it's a symm memory tensor. But it's not ideal</p> <ol> <li>there will be a lot of warnings complaining calling rendezvous on non-symm memory tensor</li> <li>I'll need to pass in the process group name to this API. But fundamentally check if a tensor is a symmetric memory tensor does not require the process group name.</li> </ol> <p>[ghstack-poisoned]</p>

In Helion autotuner, we need clone a input symm memory tensor properly if the kernel inplace update it. That requires us to know if a tensor is a symm memory tensor. Right now I call rendezvous for the tensor. If no exception is thrown, then it's a symm memory tensor. But it's not ideal

  1. there will be a lot of warnings complaining calling rendezvous on non-symm memory tensor
  2. I'll need to pass in the process group name to this API. But fundamentally check if a tensor is a symmetric memory tensor does not require the process group name.

[ghstack-poisoned]`

Assets 2

Was this article helpful?

Sign in to highlight and annotate this article

AI
Ask AI about this article
Powered by AI News Hub · full article context loaded
Ready

Conversation starters

Ask anything about this article…

Daily AI Digest

Get the top 5 AI stories delivered to your inbox every morning.

More about

update

Knowledge Map

Knowledge Map
TopicsEntitiesSource
ciflow/torc…updatePyTorch Rel…

Connected Articles — Knowledge Graph

This article is connected to other articles through shared AI topics and tags.

Knowledge Graph100 articles · 157 connections
Scroll to zoom · drag to pan · click to open

Discussion

Sign in to join the discussion

No comments yet — be the first to share your thoughts!

More in Releases