Skip to content

Commit

Permalink
Remove unroll from naive mojo variant.
Browse files Browse the repository at this point in the history
- Was added accidentally in c35a8e
  • Loading branch information
guidorice committed Dec 2, 2023
1 parent a09cbb6 commit c342441
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions mojo_impl/naive.mojo
Original file line number Diff line number Diff line change
Expand Up @@ -20,19 +20,15 @@ fn envelope[dtype: DType, dims: Int](tensor: Tensor[dtype]) -> SIMD[dtype, 2 * d
if dtype.is_floating_point():
let min_start = inf[dtype]()
let max_start = neginf[dtype]()
@unroll
for d in range(dims):
result[d] = min_start
@unroll
for d in range(dims, 2 * dims):
result[d] = max_start
else: # integral types
let min_start = max_finite[dtype]()
let max_start = min_finite[dtype]()
@unroll
for d in range(dims):
result[d] = min_start
@unroll
for d in range(dims, 2 * dims):
result[d] = max_start

Expand Down

0 comments on commit c342441

Please sign in to comment.