burn icon indicating copy to clipboard operation
burn copied to clipboard

Autotune: support int/bool ops

Open louisfd opened this issue 1 year ago • 1 comments

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)

louisfd avatar Nov 10 '23 00:11 louisfd

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

agelas avatar Dec 11 '23 07:12 agelas