Practice to write cpp/cuda extension for pytorch - View it on GitHub