diff --git a/src/layers/extensions/inference/impl.cpp b/src/layers/extensions/inference/impl.cpp
index 6419fff..9d37389 100644
--- a/src/layers/extensions/inference/impl.cpp
+++ b/src/layers/extensions/inference/impl.cpp
@@ -1,6 +1,7 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
 
+#include <torch/nn/functional.h>
 #include "def.h"
 namespace F = torch::nn::functional;