pytorch / pytorch/pytorch

TestModule: test/distributed/tensor/test_attention.py::RingAttentionTest

Open
#168,445 0 comments 0 reactions 0 assignees View on GitHub
module: ddp module: rocm module: sdpa rocm-skipped-tests triaged
Dominant language
Python
Stars
103k
Forks
29.5k
PR merge metrics
PR metrics pending

Description

## Context
- Source file: `test/distributed/tensor/test_attention.py`
- Test class: `RingAttentionTest`
- ROCm-skipped tests: 1
- Code reference: https://github.com/pytorch/pytorch/blob/main/test/distributed/tensor/test_attention.py

cc @sunway513 @jithunnair-amd @pruthvistony @ROCmSupport @jataylo @hongxiayang @naromero77amd @pragupta @jerrymannil @xinyazhang @dllehr-amd

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.