praeclarum / webgpu-torch

Tensor computation with WebGPU acceleration
MIT License
576 stars 15 forks source link

Show the GPU info in the benchmark output table #14

Open amirsojoodi opened 1 month ago

amirsojoodi commented 1 month ago

Show the GPU name/description in the column header. Sth like this:

Benchmark Time (ms) on NVIDIA GeForce GTX 1060 3GB Intel(R) Xeon(R) W-2150B CPU @ 3.00GHz NVIDIA GeForce RTX 3090 Error
unary 1d(1, 'neg') 0.013 0.001 0.003
unary 1d(1, 'sigmoid') 0.012 0.001 0.003
unary 1d(729, 'neg') 0.018 0.001 0.003
unary 1d(729, 'sigmoid') 0.010 0.002 0.003

...