Skip to content

Commit 9e5eec4

Browse files
committed
Fix compile errors
Signed-off-by: cyy <[email protected]>
1 parent edea817 commit 9e5eec4

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

kernels/portable/cpu/op_native_dropout.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
#include <executorch/kernels/portable/cpu/util/elementwise_util.h>
1010
#include <executorch/runtime/kernel/kernel_includes.h>
1111

12+
#include <optional>
1213
#include <random>
1314
#include <tuple>
1415

@@ -17,7 +18,7 @@ std::tuple<Tensor&, Tensor&> native_dropout_out(
1718
KernelRuntimeContext& ctx,
1819
const Tensor& input,
1920
double prob,
20-
torch::executor::std::optional<bool> train,
21+
::std::optional<bool> train,
2122
Tensor& out,
2223
Tensor& mask) {
2324
std::tuple<Tensor&, Tensor&> ret(out, mask);

0 commit comments

Comments
 (0)