Skip to content

[Model] Add missing operators for T5#322

Merged
yaoyaoding merged 4 commits intohidet-org:mainfrom
yaoyaoding:t5
Jul 20, 2023
Merged

[Model] Add missing operators for T5#322
yaoyaoding merged 4 commits intohidet-org:mainfrom
yaoyaoding:t5

Conversation

@yaoyaoding
Copy link
Copy Markdown
Member

  1. Add registration for torch.log and torch.abs.
  2. Use ops.full to implement ones and zeros to support symbolic shape.
  3. Add the symbolic variable name checking.

@yaoyaoding yaoyaoding merged commit bebc41b into hidet-org:main Jul 20, 2023
@yaoyaoding yaoyaoding deleted the t5 branch July 20, 2023 05:01
vadiklyutiy pushed a commit that referenced this pull request Jul 22, 2024
…322)

Similar to #281 , extend the special handling to the cases where `index`
is a tuple containing tensors, and instead of having to be on the same
device as the input tensor, each index tensor involved must (a) be on
the CPU or (b) on the same device as the input tensor.
vadiklyutiy pushed a commit that referenced this pull request Jul 23, 2024
…322)

Similar to #281 , extend the special handling to the cases where `index`
is a tuple containing tensors, and instead of having to be on the same
device as the input tensor, each index tensor involved must (a) be on
the CPU or (b) on the same device as the input tensor.
vadiklyutiy pushed a commit that referenced this pull request Dec 26, 2024
…322)

Similar to #281 , extend the special handling to the cases where `index`
is a tuple containing tensors, and instead of having to be on the same
device as the input tensor, each index tensor involved must (a) be on
the CPU or (b) on the same device as the input tensor.
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.

1 participant