RuyiAI 面向 RISC-V 架构的
人工智能系统软件栈

import torch
from buddy.compiler.frontend import dynamo_compiler

def mm(a, b):
    return torch.matmul(a, b)

a = torch.rand(2048, 2048)
b = torch.rand(2048, 2048)

compiled_mm = torch.compile(mm, backend=dynamo_compiler)
y = compiled_mm(a, b)

print(y)