|
18 | 18 | #include <iostream> |
19 | 19 |
|
20 | 20 | using namespace ::testing; |
21 | | -using exec_aten::Scalar; |
22 | | -using exec_aten::ScalarType; |
23 | | -using exec_aten::Tensor; |
| 21 | +using executorch::aten::Scalar; |
| 22 | +using executorch::aten::ScalarType; |
| 23 | +using executorch::aten::Tensor; |
| 24 | +using executorch::runtime::testing::TensorFactory; |
24 | 25 | using torch::executor::testing::SupportedFeatures; |
25 | | -using torch::executor::testing::TensorFactory; |
| 26 | +namespace etrt = executorch::runtime; |
26 | 27 |
|
27 | 28 | class OpAddOutKernelTest : public OperatorTest { |
28 | 29 | protected: |
@@ -63,7 +64,8 @@ class OpAddOutKernelTest : public OperatorTest { |
63 | 64 | test_add<DTYPE_A, DTYPE_B, ScalarType::Float>(); |
64 | 65 | test_add<DTYPE_A, DTYPE_B, ScalarType::Double>(); |
65 | 66 | // Integral out type is only allowed if both inputs are integral types |
66 | | - if (isIntegralType(DTYPE_A, false) && isIntegralType(DTYPE_B, false)) { |
| 67 | + if (etrt::isIntegralType(DTYPE_A, false) && |
| 68 | + etrt::isIntegralType(DTYPE_B, false)) { |
67 | 69 | test_add<DTYPE_A, DTYPE_B, ScalarType::Int>(); |
68 | 70 | test_add<DTYPE_A, DTYPE_B, ScalarType::Long>(); |
69 | 71 | } |
|
0 commit comments