diff --git a/src/libtorchaudio/forced_align/gpu/compute.cu b/src/libtorchaudio/forced_align/gpu/compute.cu index 52eb23bd1e..0e0686f279 100644 --- a/src/libtorchaudio/forced_align/gpu/compute.cu +++ b/src/libtorchaudio/forced_align/gpu/compute.cu @@ -1,3 +1,5 @@ +#include +#include #include #include #include diff --git a/src/libtorchaudio/iir_cuda.cu b/src/libtorchaudio/iir_cuda.cu index 2f6b75b239..7325528bfe 100644 --- a/src/libtorchaudio/iir_cuda.cu +++ b/src/libtorchaudio/iir_cuda.cu @@ -1,3 +1,5 @@ +#include +#include #include #include #include