Skip to content

Conversation

@factnn
Copy link

@factnn factnn commented Nov 20, 2025

Description

This PR enhances test coverage for the aten::index operator and fixes the AttributeError for mixed basic/advanced indexing.

Development Tool:

  • This operator was developed with Triton-Copilot, an AI-powered tool for Triton kernel development.

Changes

  • Fix AttributeError in index operator for mixed basic/advanced indexing
  • Add comprehensive test cases for index operator
  • Support combining advanced and basic indexing using Triton

Verification

  • ✅ All pytest tests pass
  • ✅ Supports mixed basic/advanced indexing (e.g., a[None, idx])
  • ✅ Comprehensive test coverage

Fixes #635

- Fix AttributeError in index operator for mixed basic/advanced indexing
- Add comprehensive test cases for index operator
- Support combining advanced and basic indexing using Triton

Fixes flagos-ai#635
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.

【operator】aten::index support

1 participant