Skip to content

[DEV] Add support for join operation in sanitizer - #178

Merged
mark14wu merged 2 commits into
mainfrom
support_create_join
Sep 26, 2025
Merged

mark14wu merged 2 commits into
mainfrom
support_create_join

Conversation

@mark14wu

Copy link
Copy Markdown
Collaborator
  • Add Join class to data.py for representing join operations
  • Import Join in patch.py and sanitizer.py
  • Add mappings for create_join in patch.py
  • Implement op_join_overrider in sanitizer to handle join operations
  • Add join to BROADCAST_OPS and create spec with proper dtype handling
  • Register Join in OP_TYPE_TO_OVERRIDER mapping
  • Add NotImplementedError for join in Z3 evaluation (to be implemented later)

- Add Join class to data.py for representing join operations
- Import Join in patch.py and sanitizer.py
- Add mappings for create_join in patch.py
- Implement op_join_overrider in sanitizer to handle join operations
- Add join to BROADCAST_OPS and create spec with proper dtype handling
- Register Join in OP_TYPE_TO_OVERRIDER mapping
- Add NotImplementedError for join in Z3 evaluation (to be implemented later)
self.dtype_tt = self.ptr.dtype_tt.element_ty


def _broadcast_dtype(self):

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

These function names better to be consistent, i.e. _dtype or _post but not mixed

@mark14wu
mark14wu merged commit 7017ed9 into main Sep 26, 2025
1 check passed
@mark14wu
mark14wu deleted the support_create_join branch September 26, 2025 16:55
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