/pytorch-extension

an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

Issues