| /* |
| * Copyright 2025 Google LLC |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| .section .data |
| .align 16 |
| |
| .global input_nothing |
| input_nothing: |
| .space 16 |
| |
| .global input_1 |
| input_1: |
| .space 16 |
| |
| .global input_2 |
| input_2: |
| .space 16 |
| |
| .global output_1 |
| output_1: |
| .space 16 |
| |
| .section .text |
| .global main |
| .option norelax |
| |
| main: |
| |
| la t1, input_1 |
| la t2, input_2 |
| la t3, output_1 |
| vsetivli x0, 16, e8, m1, ta, ma |
| |
| # loop variables |
| li t0, 1 |
| li t4, 0 |
| |
| loop8 : bge t4, t0, loop8_exit // bge t4 <= t0 loop_exit |
| |
| vle8.v v1, (t1) |
| vle8.v v2, (t2) |
| vadd.vv v3, v1, v2 |
| addi t1, t1, 16 |
| addi t2, t2, 16 |
| addi t3, t3, 16 |
| addi t4, t4, 1 |
| j loop8 |
| loop8_exit: |
| wfi |
| |
| la t1, input_1 |
| la t2, input_2 |
| la t3, output_1 |
| vsetivli x0, 8, e16, m1, ta, ma |
| # loop variables |
| li t0, 1 |
| li t4, 0 |
| |
| loop16 : bge t4, t0, loop16_exit |
| vle8.v v1, (t1) |
| vle8.v v2, (t2) |
| vadd.vv v3, v1, v2 |
| addi t1, t1, 8 |
| addi t2, t2, 8 |
| addi t3, t3, 8 |
| addi t4, t4, 1 |
| j loop16 |
| loop16_exit: |
| wfi |
| |
| |
| la t1, input_1 |
| la t2, input_2 |
| la t3, output_1 |
| vsetivli x0, 4, e32, m1, ta, ma |
| # loop variables |
| li t0, 1 |
| li t4, 0 |
| |
| loop32 : bge t4, t0, loop32_exit |
| vle8.v v1, (t1) |
| vle8.v v2, (t2) |
| vadd.vv v3, v1, v2 |
| addi t1, t1, 4 |
| addi t2, t2, 4 |
| addi t3, t3, 4 |
| addi t4, t4, 1 |
| j loop32 |
| loop32_exit: |
| wfi |
| ret |
