search

facebookincubator / AITemplate

AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.
Apache License 2.0
4.54k stars 363 forks source link

ait: Explicitly throw when indexing a boolean tensor for masking #992

Closed kflu closed 7 months ago

kflu commented 7 months ago

Summary: Currently AIT hasn't implemented tensor[boolean_tensor] for masking. It fail shortly after this call, at:

'Tensor' object has no attribute 'upper_bound'

> link-tree/aitemplate/utils/shape_utils.py(195)convert_IntVar_to_int()
-> if var.upper_bound() == var.lower_bound():

Reviewed By: khabinov

Differential Revision: D53654054

facebook-github-bot commented 7 months ago

This pull request was exported from Phabricator. Differential Revision: D53654054

facebook-github-bot commented 7 months ago

This pull request was exported from Phabricator. Differential Revision: D53654054

facebook-github-bot commented 7 months ago

This pull request was exported from Phabricator. Differential Revision: D53654054

facebook-github-bot commented 7 months ago

This pull request was exported from Phabricator. Differential Revision: D53654054

facebook-github-bot commented 7 months ago

This pull request was exported from Phabricator. Differential Revision: D53654054

facebook-github-bot commented 7 months ago

This pull request was exported from Phabricator. Differential Revision: D53654054

facebook-github-bot commented 7 months ago

This pull request has been merged in facebookincubator/AITemplate@a99c753c227a44fff821097e0f8fc548cf567f16.