[FEA]: Random Number Generation

[tomasruizt]

Is this a new feature, an improvement, or a change to existing functionality?

New Feature

How would you describe the priority of this feature request?

Critical (currently preventing usage)

Please provide a clear description of problem this feature solves

I searched in the documentation for random number generation functions but have not found any so far.

As a reference, see Triton's API: https://triton-lang.org/main/python-api/triton.language.html#random-number-generation

Feature Description

One usage example is to create noise shaped with the size of the tiles. E.g for gumbel-max sampling.

Describe your ideal solution

The Triton API is a good starting point.

Describe any alternatives you have considered

No response

Additional context

No response

Contributing Guidelines

• I agree to follow cuTile Python's contributing guidelines
• I have searched the open feature requests and have found no duplicates for this feature request

[HiralThadeshwar31]

@tomasruizt Could you assign me this issue?
Thanks,
Hiral

[tomasruizt]

I don't own the repository and I don't have the authority to assign issues exclusively to people. @HiralThadeshwar31

[haijieg]

@tomasruizt thank you for your interest. This feature will requires CUDA TileIR support. Please stay in tuned.