Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Autotune: support int/bool ops #944

Open
louisfd opened this issue Nov 10, 2023 · 1 comment
Open

Autotune: support int/bool ops #944

louisfd opened this issue Nov 10, 2023 · 1 comment
Labels
enhancement Enhance existing features

Comments

@louisfd
Copy link
Member

louisfd commented Nov 10, 2023

In autotune, we create random tensors as inputs. Since random only works with floats it makes it impossible to use autotune with int and bool.

Lazy solution: revert to creating ones tensor instead of random
Hacky solution: Run the random kernel on f32 instead of E then cast to int or bool using some
Long solution: Add support at large for bool random (bernoulli) and int random (uniform)

Once this is done we can add a call to autotune for sum_dim and mean_dim (int versions)

@antimora antimora added the enhancement Enhance existing features label Nov 19, 2023
@agelas
Copy link
Contributor

agelas commented Dec 11, 2023

@antimora I think I'll have some more free time coming up, I can tentatively take this.

@agelas agelas mentioned this issue Jan 12, 2024
2 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement Enhance existing features
Projects
None yet
Development

No branches or pull requests

3 participants