Skip to content

Comments

Add torch sdpa backend to ssim test#316

Merged
SolitaryThinker merged 4 commits intomainfrom
will/ssim-test-add-sdpa
Apr 7, 2025
Merged

Add torch sdpa backend to ssim test#316
SolitaryThinker merged 4 commits intomainfrom
will/ssim-test-add-sdpa

Conversation

@SolitaryThinker
Copy link
Collaborator

No description provided.

@pytest.mark.parametrize("num_inference_steps", [6])
@pytest.mark.parametrize("prompt", TEST_PROMPTS)
def test_inference_similarity(num_inference_steps, prompt):
@pytest.mark.parametrize("ATTENTION_BACKEND", ["FLASH_ATTN"])
Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@kevin314 Please add "TORCH_SDPA" to this list once you add the reference video for A40

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

And then please merge if it passes CI

@kevin314 kevin314 temporarily deployed to runpod-runners April 7, 2025 01:34 — with GitHub Actions Inactive
@kevin314 kevin314 temporarily deployed to runpod-runners April 7, 2025 01:48 — with GitHub Actions Inactive
@SolitaryThinker SolitaryThinker self-assigned this Apr 7, 2025
@SolitaryThinker SolitaryThinker merged commit c64f826 into main Apr 7, 2025
4 of 8 checks passed
@SolitaryThinker SolitaryThinker deleted the will/ssim-test-add-sdpa branch April 7, 2025 22:42
kevin314 added a commit to kevin314/FastVideo that referenced this pull request Apr 8, 2025
Co-authored-by: kevin314 <kevin.lin.cs1@gmail.com>
qimcis pushed a commit to qimcis/FastVideo that referenced this pull request Oct 30, 2025
Co-authored-by: kevin314 <kevin.lin.cs1@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants