feat(dnn): add indexing_one_hot and indexing_set_one_hot opr

GitOrigin-RevId: c5406c71ff

dnn/test/common/opr_trait.h → dnn/src/common/opr_trait.h

@@ -1,5 +1,5 @@
 /**
- * \file dnn/test/common/opr_trait.h
+ * \file dnn/src/common/opr_trait.h
  * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
  *
  * Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
@@ -14,7 +14,6 @@
 #include <cstddef>
 
 namespace megdnn {
-namespace test {
 
 template <typename Opr>
 struct OprTrait {};
@@ -114,7 +113,10 @@ DEF(FakeQuantForward, 4, true, true);
 DEF(FakeQuantBackward, 5, true, false);
 DEF(TQTForward, 3, true, true);
 DEF(TQTBackward, 5, true, false);
-}  // namespace test
+DEF(PowC, 2, false, true);
+DEF(UniformRNG, 1, true, true);
+DEF(GaussianRNG, 1, true, true);
+
 }  // namespace megdnn
 
 // vim: syntax=cpp.doxygen

dnn/test/common/opr_algo_proxy.h

@@ -12,7 +12,7 @@
 #pragma once
 
 #include "megdnn/basic_types.h"
-#include "test/common/opr_trait.h"
+#include "src/common/opr_trait.h"
 #include "test/common/utils.h"
 
 namespace megdnn {

dnn/test/common/opr_proxy.h

@@ -11,12 +11,13 @@
  */
 #pragma once
 
+#include "src/common/opr_trait.h"
+
 #include "test/common/deduce_layout_proxy.h"
 #include "test/common/exec_proxy.h"
 #include "test/common/fast_run_cache.h"
 #include "test/common/inspect_type.h"
 #include "test/common/opr_algo_proxy.h"
-#include "test/common/opr_trait.h"
 #include "test/common/timer.h"
 #include "test/common/workspace_wrapper.h"
 

dnn/test/common/powc.h

@@ -12,13 +12,12 @@
 
 #include "megdnn/handle.h"
 #include "megdnn/oprs/general.h"
-#include "test/common/opr_proxy.h"
+
+#include "src/common/opr_trait.h"
 
 namespace megdnn {
 namespace test {
 
-DEF(PowC, 2, false, true);
-
 void run_powc_test(Handle* handle, DType dtype);
 
 }  // namespace test