ModularM
Modular17mo ago
1 reply
Opschietto

Conditional parameter evaluation

Hello, I was wondering if there was a way to write conditional parameter evaluation, something like @parameter(condition)
to apply the @parameter only if condition is true.
Practial example of unrolling loop:

# Current:
@parameter
if nchunks <= 4:
    @parameter
    for chunk in range(nchunks):
        var idx = ...
        dest.store(idx, src.load[width=32](idx))
else:
    for chunk in range(nchunks):
        var idx = ...
        dest.store(idx, src.load[width=32](idx))

# Ideal:
@parameter(nchunks <= 4)
for chunk in range(nchunks):
    var idx = ...
    dest.store(idx, src.load[width=32](idx))
Was this page helpful?