Add native/quantized to the list of header rewrites (#31151)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/31151

same as title. I am not sure why this was not added in the first place.

Test Plan: wait for build to succeed.

Reviewed By: bddppq, xw285cornell

Differential Revision: D18880216

fbshipit-source-id: 8b17d4fbd5dd08c28c52df8b1da77b69d56d65dc
This commit is contained in:
Serhat Yilmaz
2019-12-11 21:57:24 -08:00
committed by Facebook Github Bot
parent 6ab2d1b1a4
commit 4f5a4be45f

View File

@ -632,6 +632,7 @@ def preprocessor(output_directory, filepath, stats, hip_clang_launch):
if (
f.startswith("ATen/cuda")
or f.startswith("ATen/native/cuda")
or f.startswith("ATen/native/quantized/cuda")
or f.startswith("ATen/native/sparse/cuda")
or f.startswith("THC/")
or f.startswith("THCUNN/")