chn-lee-yumi

import importlib

import torch


def test_supported_dtypes(device):
    dtypes = [
        torch.bool,
        torch.uint8,
        torch.int8,

chn-lee-yumi

import importlib
import time

import torch


def benchmark_dtype(dtype, device, size, repeat):
    try:
        # 创建两个张量
        a = torch.randn((size, size), dtype=dtype, device=device)

chn-lee-yumi

package main

// go version of https://gist.github.com/javiermon/43a0b9e1c07abd4b13df

import (
	"log"
	"github.com/jsimonetti/rtnetlink"
	"fmt"
)