Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
yaolubrain authored Jul 4, 2024
1 parent 6522f86 commit cd3309e
Showing 1 changed file with 13 additions and 0 deletions.
13 changes: 13 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,3 +1,16 @@
# 4D Convolution and Cost Volume in CUDA

A simple 4D convolution layer and 4D cost volume layer in CUDA with Pytorch interface.

Compile and test

```
cd cpp
python setup.py install
python test.py
cd ..
cd cuda
python setup.py install
python test.py
```

0 comments on commit cd3309e

Please sign in to comment.