cleanup
This commit is contained in:
parent
b6b3973b3b
commit
494b22fbbd
|
@ -17,17 +17,19 @@ 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
|
||||||
|
|
||||||
|
CUDA 11.1
|
||||||
|
CuDNN 8.0.5
|
||||||
```
|
```
|
||||||
|
|
||||||
## `gotch` conv2d benchmark
|
## `gotch`
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
name time/op
|
name time/op
|
||||||
Conv2dCPU-8 21.2ms ± 0%
|
Conv2dCPU-8 21.2ms ± 0%
|
||||||
Conv2dCUDA-8 2.20ms ± 0%
|
Conv2dCUDA-8 2.20ms ± 0%
|
||||||
```
|
```
|
||||||
|
|
||||||
## Python `Pytorch` 1.11 CUDA 11 benchmark
|
## Python `Pytorch` 1.11
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
conv2d-CPU(x): 56.7 ms
|
conv2d-CPU(x): 56.7 ms
|
||||||
|
@ -38,7 +40,6 @@ benchmark Python code below
|
||||||
|
|
||||||
```python
|
```python
|
||||||
import torch
|
import torch
|
||||||
import torch.nn.functional as F
|
|
||||||
import timeit
|
import timeit
|
||||||
|
|
||||||
x = torch.randn(32, 64, 64, 64)
|
x = torch.randn(32, 64, 64, 64)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user