Searched defs:round_multiple (Results 1 – 3 of 3) sorted by relevance
442 auto round_multiple = [](int x, int m) { return (x + m - 1) / m * m; }; in mha_fwd() local681 auto round_multiple = [](int x, int m) { return (x + m - 1) / m * m; }; in mha_varlen_fwd() local880 auto round_multiple = [](int x, int m) { return (x + m - 1) / m * m; }; in mha_bwd() local1102 auto round_multiple = [](int x, int m) { return (x + m - 1) / m * m; }; in mha_varlen_bwd() local1377 auto round_multiple = [](int x, int m) { return (x + m - 1) / m * m; }; in mha_fwd_kvcache() local
134 static unsigned int round_multiple(unsigned int x, unsigned int multiple) in round_multiple() function
133 static unsigned int round_multiple(unsigned int x, unsigned int multiple) in round_multiple() function