Derived constexprs

[davisyoshida]

Is it possible to create a new constexpr from an existing one? E.g. if I have a constant BLOCK_SIZE, I might want to calculate HALF_BLOCK_SIZE = BLOCK_SIZE // 2. However, if I do this HALF_BLOCK_SIZE is not accepted as an upper limit for tl.arange.

Is it necessary to just repeat the expression BLOCK_SIZE // 2 everywhere I need the derived value? Or is there some way to create a constant for it?

[ptillet]

Yes. You need to annotate the variables with ': tl.constexpr' in the kennel

[davisyoshida]

Thanks! I'm not used to type annotations having any effect at runtime still whoops.

[ptillet]

Tbh it should be better documented :)