andr3/gotch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

improved ts/README.md

Browse Source
This commit is contained in:
sugarme 2022-05-06 18:16:32 +10:00
parent 670d1e9cdf
commit b6b3973b3b
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ts/README.md
View File

@ -10,12 +10,16 @@ Benchmark tensor operation `conv2d` forward propagation:
- input shape: `[32, 64, 64, 64]` - input shape: `[32, 64, 64, 64]`
- kernel: `[64, 3, 3]` - kernel: `[64, 3, 3]`
```bash
goos: linux goos: linux
goarch: amd64 goarch: amd64
pkg: github.com/sugarme/gotch/ts pkg: github.com/sugarme/gotch/ts
cpu: Intel(R) Core(TM) i7-6700HQ CPU @ 2.60GHz cpu: Intel(R) Core(TM) i7-6700HQ CPU @ 2.60GHz
BenchmarkConv2dCPU-8 100 21198303 ns/op BenchmarkConv2dCPU-8 100 21198303 ns/op
BenchmarkConv2dCUDA-8 100 2201213 ns/op BenchmarkConv2dCUDA-8 100 2201213 ns/op
```
## `gotch` conv2d benchmark
```bash ```bash
name time/op name time/op
@ -23,7 +27,7 @@ Conv2dCPU-8 21.2ms ± 0%
Conv2dCUDA-8 2.20ms ± 0% Conv2dCUDA-8 2.20ms ± 0%
``` ```
Benchmark against Pytorch 1.11 CUDA 11 ## Python `Pytorch` 1.11 CUDA 11 benchmark
```bash ```bash
conv2d-CPU(x): 56.7 ms conv2d-CPU(x): 56.7 ms