pytorch
93a53ff4 - Create a quantized version ReLU function for CUDA (#85502)

Commit
3 years ago
Create a quantized version ReLU function for CUDA (#85502) Summary: this is to allow the relu function to run on a quantized tensor on cuda. That is torch.relu(qa) for a quantized tensor qa on cuda. Test Plan: python test/test_quantization.py Pull Request resolved: https://github.com/pytorch/pytorch/pull/85502 Approved by: https://github.com/dzdang
Author
Committer
Parents
Loading