├── .gitmodules ├── CMakeLists.txt ├── LICENSE ├── README-build.md ├── README.md ├── benchmark.md ├── benchmark_eval.sh ├── benchmark_train.sh ├── benchmarks └── v0.2.0 │ ├── summary.md │ ├── test-A380-opencl-batch32.txt │ ├── test-A380-xpu-batch32.txt │ ├── test-gtx960-cuda-batch32.txt │ ├── test-gtx960-opencl-batch32.txt │ ├── test-rx6600-opencl-batch64.txt │ ├── test-rx6600-rocm-batch64.txt │ ├── train-A380-opencl-small.txt │ ├── train-A380-xpu-small.txt │ ├── train-gtx960-cuda-small.txt │ ├── train-gtx960-opencl-small.txt │ ├── train-rx6600-opencl-large.txt │ └── train-rx6600-rocm-large.txt ├── examples └── cplus │ ├── CMakeLists.txt │ └── infer_mobilenet │ └── infer_mobilenet.cpp ├── mnist.py ├── mnist_cnn-scripted.pt ├── python └── pytorch_ocl │ └── __init__.py ├── src ├── CLTensor.cpp ├── CLTensor.h ├── guard.cpp ├── loss_ops.cpp ├── norm_ops.cpp ├── pointwise_ops.cpp ├── python_iface.cpp ├── random_ops.cpp ├── registeration.cpp ├── tensor_ops.cpp ├── utils.cpp ├── utils.h └── vision_ops.cpp ├── test.py ├── tests ├── big_samples │ ├── cat.ppm │ ├── dog.ppm │ ├── goldfish.ppm │ └── parrot.ppm ├── samples │ ├── cat.ppm │ ├── dog.ppm │ ├── goldfish.ppm │ └── parrot.ppm ├── small_samples │ ├── cat.ppm │ ├── dog.ppm │ ├── goldfish.ppm │ └── parrot.ppm ├── test_op.py ├── test_print.py └── validate_network.py └── version.txt /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/.gitmodules -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/LICENSE -------------------------------------------------------------------------------- /README-build.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/README-build.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/README.md -------------------------------------------------------------------------------- /benchmark.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/benchmark.md -------------------------------------------------------------------------------- /benchmark_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/benchmark_eval.sh -------------------------------------------------------------------------------- /benchmark_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/benchmark_train.sh -------------------------------------------------------------------------------- /benchmarks/v0.2.0/summary.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/benchmarks/v0.2.0/summary.md -------------------------------------------------------------------------------- /benchmarks/v0.2.0/test-A380-opencl-batch32.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/benchmarks/v0.2.0/test-A380-opencl-batch32.txt -------------------------------------------------------------------------------- /benchmarks/v0.2.0/test-A380-xpu-batch32.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/benchmarks/v0.2.0/test-A380-xpu-batch32.txt -------------------------------------------------------------------------------- /benchmarks/v0.2.0/test-gtx960-cuda-batch32.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/benchmarks/v0.2.0/test-gtx960-cuda-batch32.txt -------------------------------------------------------------------------------- /benchmarks/v0.2.0/test-gtx960-opencl-batch32.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/benchmarks/v0.2.0/test-gtx960-opencl-batch32.txt -------------------------------------------------------------------------------- /benchmarks/v0.2.0/test-rx6600-opencl-batch64.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/benchmarks/v0.2.0/test-rx6600-opencl-batch64.txt -------------------------------------------------------------------------------- /benchmarks/v0.2.0/test-rx6600-rocm-batch64.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/benchmarks/v0.2.0/test-rx6600-rocm-batch64.txt -------------------------------------------------------------------------------- /benchmarks/v0.2.0/train-A380-opencl-small.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/benchmarks/v0.2.0/train-A380-opencl-small.txt -------------------------------------------------------------------------------- /benchmarks/v0.2.0/train-A380-xpu-small.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/benchmarks/v0.2.0/train-A380-xpu-small.txt -------------------------------------------------------------------------------- /benchmarks/v0.2.0/train-gtx960-cuda-small.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/benchmarks/v0.2.0/train-gtx960-cuda-small.txt -------------------------------------------------------------------------------- /benchmarks/v0.2.0/train-gtx960-opencl-small.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/benchmarks/v0.2.0/train-gtx960-opencl-small.txt -------------------------------------------------------------------------------- /benchmarks/v0.2.0/train-rx6600-opencl-large.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/benchmarks/v0.2.0/train-rx6600-opencl-large.txt -------------------------------------------------------------------------------- /benchmarks/v0.2.0/train-rx6600-rocm-large.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/benchmarks/v0.2.0/train-rx6600-rocm-large.txt -------------------------------------------------------------------------------- /examples/cplus/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/examples/cplus/CMakeLists.txt -------------------------------------------------------------------------------- /examples/cplus/infer_mobilenet/infer_mobilenet.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/examples/cplus/infer_mobilenet/infer_mobilenet.cpp -------------------------------------------------------------------------------- /mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/mnist.py -------------------------------------------------------------------------------- /mnist_cnn-scripted.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/mnist_cnn-scripted.pt -------------------------------------------------------------------------------- /python/pytorch_ocl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/python/pytorch_ocl/__init__.py -------------------------------------------------------------------------------- /src/CLTensor.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/src/CLTensor.cpp -------------------------------------------------------------------------------- /src/CLTensor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/src/CLTensor.h -------------------------------------------------------------------------------- /src/guard.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/src/guard.cpp -------------------------------------------------------------------------------- /src/loss_ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/src/loss_ops.cpp -------------------------------------------------------------------------------- /src/norm_ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/src/norm_ops.cpp -------------------------------------------------------------------------------- /src/pointwise_ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/src/pointwise_ops.cpp -------------------------------------------------------------------------------- /src/python_iface.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/src/python_iface.cpp -------------------------------------------------------------------------------- /src/random_ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/src/random_ops.cpp -------------------------------------------------------------------------------- /src/registeration.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/src/registeration.cpp -------------------------------------------------------------------------------- /src/tensor_ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/src/tensor_ops.cpp -------------------------------------------------------------------------------- /src/utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/src/utils.cpp -------------------------------------------------------------------------------- /src/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/src/utils.h -------------------------------------------------------------------------------- /src/vision_ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/src/vision_ops.cpp -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/test.py -------------------------------------------------------------------------------- /tests/big_samples/cat.ppm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/tests/big_samples/cat.ppm -------------------------------------------------------------------------------- /tests/big_samples/dog.ppm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/tests/big_samples/dog.ppm -------------------------------------------------------------------------------- /tests/big_samples/goldfish.ppm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/tests/big_samples/goldfish.ppm -------------------------------------------------------------------------------- /tests/big_samples/parrot.ppm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/tests/big_samples/parrot.ppm -------------------------------------------------------------------------------- /tests/samples/cat.ppm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/tests/samples/cat.ppm -------------------------------------------------------------------------------- /tests/samples/dog.ppm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/tests/samples/dog.ppm -------------------------------------------------------------------------------- /tests/samples/goldfish.ppm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/tests/samples/goldfish.ppm -------------------------------------------------------------------------------- /tests/samples/parrot.ppm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/tests/samples/parrot.ppm -------------------------------------------------------------------------------- /tests/small_samples/cat.ppm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/tests/small_samples/cat.ppm -------------------------------------------------------------------------------- /tests/small_samples/dog.ppm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/tests/small_samples/dog.ppm -------------------------------------------------------------------------------- /tests/small_samples/goldfish.ppm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/tests/small_samples/goldfish.ppm -------------------------------------------------------------------------------- /tests/small_samples/parrot.ppm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/tests/small_samples/parrot.ppm -------------------------------------------------------------------------------- /tests/test_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/tests/test_op.py -------------------------------------------------------------------------------- /tests/test_print.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/tests/test_print.py -------------------------------------------------------------------------------- /tests/validate_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/artyom-beilis/pytorch_dlprim/HEAD/tests/validate_network.py -------------------------------------------------------------------------------- /version.txt: -------------------------------------------------------------------------------- 1 | 0.2.0 2 | --------------------------------------------------------------------------------