Mojo module
grouped_matmul
Aliases
NumWarpPerWarpGroup
alias NumWarpPerWarpGroup = 0 if (_resolve_warp_size() == 0) else (div_s(#lit.struct.extract<:@stdlib::@builtin::@int::@Int apply(:!lit.generator<() -> !lit.struct<@stdlib::@builtin::@int::@Int>> @stdlib::@gpu::@globals::@"_resolve_warpgroup_size()"), "value">, #lit.struct.extract<:@stdlib::@builtin::@int::@Int cond(eq(#lit.struct.extract<:@stdlib::@builtin::@int::@Int apply(:!lit.generator<() -> !lit.struct<@stdlib::@builtin::@int::@Int>> @stdlib::@gpu::@globals::@"_resolve_warp_size()"), "value">, 0), {1}, apply(:!lit.generator<() -> !lit.struct<@stdlib::@builtin::@int::@Int>> @stdlib::@gpu::@globals::@"_resolve_warp_size()")), "value">) + -1) if (((rem_s(#lit.struct.extract<:@stdlib::@builtin::@int::@Int apply(:!lit.generator<() -> !lit.struct<@stdlib::@builtin::@int::@Int>> @stdlib::@gpu::@globals::@"_resolve_warpgroup_size()"), "value">, #lit.struct.extract<:@stdlib::@builtin::@int::@Int cond(eq(#lit.struct.extract<:@stdlib::@builtin::@int::@Int apply(:!lit.generator<() -> !lit.struct<@stdlib::@builtin::@int::@Int>> @stdlib::@gpu::@globals::@"_resolve_warp_size()"), "value">, 0), {1}, apply(:!lit.generator<() -> !lit.struct<@stdlib::@builtin::@int::@Int>> @stdlib::@gpu::@globals::@"_resolve_warp_size()")), "value">) == 0) ^ True) & ((_resolve_warp_size() < 0) ^ (_resolve_warpgroup_size() < 0))) else div_s(#lit.struct.extract<:@stdlib::@builtin::@int::@Int apply(:!lit.generator<() -> !lit.struct<@stdlib::@builtin::@int::@Int>> @stdlib::@gpu::@globals::@"_resolve_warpgroup_size()"), "value">, #lit.struct.extract<:@stdlib::@builtin::@int::@Int cond(eq(#lit.struct.extract<:@stdlib::@builtin::@int::@Int apply(:!lit.generator<() -> !lit.struct<@stdlib::@builtin::@int::@Int>> @stdlib::@gpu::@globals::@"_resolve_warp_size()"), "value">, 0), {1}, apply(:!lit.generator<() -> !lit.struct<@stdlib::@builtin::@int::@Int>> @stdlib::@gpu::@globals::@"_resolve_warp_size()")), "value">)
Functions
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!