DaCe works on RISC-V out of the box!

[computablee]

After spending a summer with DaCe, I was curious how well it would end up working on a RISC-V system. @tbennun let me know that the source should be portable and that it should work on RISC-V, but I wanted to see it in action.

I finally found the time today to try it out, and sure enough, DaCe is so portable that it runs seamlessly on my StarFive VisionFive2 development board.

screenfetch output:

         _,met$$$$$gg.           lane@starfive
      ,g$$$$$$$$$$$$$$$P.        OS: Debian  trixie
    ,g$$P""       """Y$$.".      Kernel: riscv64 Linux 5.15.0-starfive
   ,$$P'              `$$$.      Uptime: 1h 20m
  ',$$P       ,ggs.     `$$b:    Packages: 1220
  `d$$'     ,$P"'   .    $$$     Shell: bash 5.2.15
   $$P      d$'     ,    $$P     Disk: 8.6G / 242G (4%)
   $$:      $$.   -    ,d$$'     CPU: Unknown @ 4x 1.5GHz
   $$\;      Y$b._   _,d$P'      RAM: 531MiB / 7925MiB
   Y$$.    `.`"Y$$$$P"'         
   `$$b      "-.__              
    `Y$$                        
     `Y$$.                      
       `$$b.                    
         `Y$$b.                 
            `"Y$b._             
                `""""           

lscpu output:

Architecture:          riscv64
  Byte Order:          Little Endian
CPU(s):                4
  On-line CPU(s) list: 0-3

DaCe test:

(dace_env) lane@starfive:~/programs/dace_test$ cat test.py 
import dace
import numpy as np

N = 10000

@dace.program
def add(x: dace.float32[N], y: dace.float32[N]):
    z = dace.ndarray(N, dtype=dace.float32)

    for i in dace.map[0:N]:
        z[i] = x[i] + y[i]

    return z

a = np.random.rand(N).astype(np.float32)
b = np.random.rand(N).astype(np.float32)

sdfg = add.to_sdfg()

c1 = a + b
c2 = sdfg(x=a, y=b)

print(np.all(c1 == c2))

(dace_env) lane@starfive:~/programs/dace_test$ python3 test.py 
True

It ended up being far more of a nuisance getting all the dependencies installed (OpenBLAS, Numpy, Pip, etc.) but once all that was installed, the only DaCe-specific thing I had to change was using -march=rv64gc instead of -march=native in my .dace.conf.

Not exactly sure where else a good place to post this would be, so I figured I'd post it here! Huge kudos to the contributors.