/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/core/utility/philox_rand.hpp Source File#
philox_rand.hpp
Go to the documentation of this file.
10 // Reference: https://github.com/Dao-AILab/flash-attention/blob/main/csrc/flash_attn/src/philox.cuh
Definition: philox_rand.hpp:12
CK_TILE_HOST_DEVICE uint4 get_philox_4x32(const unsigned long long subsequence) const
Definition: philox_rand.hpp:22
CK_TILE_HOST_DEVICE void get_random_4x8(uint8_t *out, const unsigned long long subsequence, const index_t idx) const
Definition: philox_rand.hpp:75
CK_TILE_HOST_DEVICE philox(unsigned long long seed_, unsigned long long offset_)
Definition: philox_rand.hpp:14
CK_TILE_HOST_DEVICE void get_random_8x8(uint8_t *out, const unsigned long long subsequence, const index_t idx0, const index_t idx1) const
Definition: philox_rand.hpp:56
CK_TILE_HOST_DEVICE void get_random_16x8(uint8_t *out, const unsigned long long subsequence) const
Definition: philox_rand.hpp:42
Definition: cluster_descriptor.hpp:13