OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: 98b5ee78c1
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '98b5ee78c1'
${ noResults }
MegEngine/dnn/src/naive/handle.cpp

145 lines
5.8 kB
Raw Blame History 

  1. #include "src/naive/handle.h"

  2. 

  3. #include "src/common/handle_impl.h"

  4. 

  5. #include "src/naive/adaptive_pooling/opr_impl.h"

  6. #include "src/naive/add_update/opr_impl.h"

  7. #include "src/naive/argmxx/opr_impl.h"

  8. #include "src/naive/argsort/opr_impl.h"

  9. #include "src/naive/batch_conv_bias/opr_impl.h"

  10. #include "src/naive/batch_normalization/opr_impl.h"

  11. #include "src/naive/batched_matrix_mul/opr_impl.h"

  12. #include "src/naive/check_non_finite/opr_impl.h"

  13. #include "src/naive/checksum/opr_impl.h"

  14. #include "src/naive/concat/opr_impl.h"

  15. #include "src/naive/cond_take/opr_impl.h"

  16. #include "src/naive/conv_bias/opr_impl.h"

  17. #include "src/naive/convolution/opr_impl.h"

  18. #include "src/naive/convolution3d/opr_impl.h"

  19. #include "src/naive/convpooling/opr_impl.h"

  20. #include "src/naive/correlation/opr_impl.h"

  21. #include "src/naive/cumsum/opr_impl.h"

  22. #include "src/naive/cvt_color/opr_impl.h"

  23. #include "src/naive/dct/opr_impl.h"

  24. #include "src/naive/deformable_conv/opr_impl.h"

  25. #include "src/naive/deformable_ps_roi_pooling/opr_impl.h"

  26. #include "src/naive/diag/opr_impl.h"

  27. #include "src/naive/dot/opr_impl.h"

  28. #include "src/naive/dropout/opr_impl.h"

  29. #include "src/naive/elemwise/opr_impl.h"

  30. #include "src/naive/elemwise_multi_type/opr_impl.h"

  31. #include "src/naive/eye/opr_impl.h"

  32. #include "src/naive/fake_quant/opr_impl.h"

  33. #include "src/naive/fill/opr_impl.h"

  34. #include "src/naive/flip/opr_impl.h"

  35. #include "src/naive/gaussian_blur/opr_impl.h"

  36. #include "src/naive/group_local/opr_impl.h"

  37. #include "src/naive/images2neibs/opr_impl.h"

  38. #include "src/naive/indexing_multi_axis_vec/opr_impl.h"

  39. #include "src/naive/indexing_one_hot/opr_impl.h"

  40. #include "src/naive/lamb/opr_impl.h"

  41. #include "src/naive/layer_norm/opr_impl.h"

  42. #include "src/naive/linspace/opr_impl.h"

  43. #include "src/naive/local/opr_impl.h"

  44. #include "src/naive/local_share/opr_impl.h"

  45. #include "src/naive/lrn/opr_impl.h"

  46. #include "src/naive/lsq/opr_impl.h"

  47. #include "src/naive/lstm/opr_impl.h"

  48. #include "src/naive/lstm_cell/opr_impl.h"

  49. #include "src/naive/mask_conv/opr_impl.h"

  50. #include "src/naive/matrix_inverse/opr_impl.h"

  51. #include "src/naive/matrix_mul/opr_impl.h"

  52. #include "src/naive/max_tensor_diff/opr_impl.h"

  53. #include "src/naive/mesh_indexing/opr_impl.h"

  54. #include "src/naive/padding/opr_impl.h"

  55. #include "src/naive/param_pack/opr_impl.h"

  56. #include "src/naive/pooling/opr_impl.h"

  57. #include "src/naive/powc/opr_impl.h"

  58. #include "src/naive/reduce/opr_impl.h"

  59. #include "src/naive/relayout/opr_impl.h"

  60. #include "src/naive/relayout_format/opr_impl.h"

  61. #include "src/naive/remap/opr_impl.h"

  62. #include "src/naive/repeat/opr_impl.h"

  63. #include "src/naive/resize/opr_impl.h"

  64. #include "src/naive/rng/opr_impl.h"

  65. #include "src/naive/rnn/opr_impl.h"

  66. #include "src/naive/rnn_cell/opr_impl.h"

  67. #include "src/naive/roi_align/opr_impl.h"

  68. #include "src/naive/roi_copy/opr_impl.h"

  69. #include "src/naive/roi_pooling/opr_impl.h"

  70. #include "src/naive/rotate/opr_impl.h"

  71. #include "src/naive/separable_conv/opr_impl.h"

  72. #include "src/naive/separable_filter/opr_impl.h"

  73. #include "src/naive/sleep/opr_impl.h"

  74. #include "src/naive/sliding_window_transpose/opr_impl.h"

  75. #include "src/naive/softmax/opr_impl.h"

  76. #include "src/naive/split/opr_impl.h"

  77. #include "src/naive/svd/opr_impl.h"

  78. #include "src/naive/tensor_remap/opr_impl.h"

  79. #include "src/naive/tile/opr_impl.h"

  80. #include "src/naive/topk/opr_impl.h"

  81. #include "src/naive/tqt/opr_impl.h"

  82. #include "src/naive/transpose/opr_impl.h"

  83. #include "src/naive/type_cvt/opr_impl.h"

  84. #include "src/naive/warp_affine/opr_impl.h"

  85. #include "src/naive/warp_perspective/opr_impl.h"

  86. 

  87. namespace megdnn {

  88. namespace naive {

  89. //! always for ci

  90. static size_t g_image2d_pitch_alignment = 1;

  91. static HandleImpl::HandleVendorType g_image2d_pitch_vendor =

  92.         HandleImpl::HandleVendorType::NOT_SPEC;

  93. 

  94. DefaultConvolutionForwardAlgorithm HandleImpl::m_default_conv_fwd_algo;

  95. DefaultConvolutionBackwardDataAlgorithm HandleImpl::m_default_conv_bwd_data_algo;

  96. DefaultConvolutionBackwardFilterAlgorithm HandleImpl::m_default_conv_bwd_filter_algo;

  97. DefaultConvBiasForwardAlgorithm HandleImpl::m_default_conv_bias_fwd_algo;

  98. DefaultConvolution3DForwardAlgorithm HandleImpl::m_default_conv3d_fwd_algo;

  99. DefaultConvolution3DBackwardDataAlgorithm HandleImpl::m_default_conv3d_bwd_data_algo;

  100. DefaultConvolution3DBackwardFilterAlgorithm

  101.         HandleImpl::m_default_conv3d_bwd_filter_algo;

  102. DefaultBatchConvBiasForwardAlgorithm HandleImpl::m_default_batch_conv_bias_fwd_algo;

  103. DefaultLocalShareForwardAlgorithm HandleImpl::m_default_local_share_fwd_algo;

  104. DefaultLocalShareBackwardDataAlgorithm HandleImpl::m_default_local_share_bwd_data_algo;

  105. DefaultLocalShareBackwardFilterAlgorithm

  106.         HandleImpl::m_default_local_share_bwd_filter_algo;

  107. 

  108. DefaultMatrixMulAlgorithm HandleImpl::m_default_matmul_fwd_algo;

  109. DefaultBatchedMatrixMulAlgorithm HandleImpl::m_default_batched_matmul_fwd_algo;

  110. 

  111. DefaultPoolingForwardAlgorithm HandleImpl::m_default_pooling_fwd_algo;

  112. DefaultPoolingBackwardAlgorithm HandleImpl::m_default_pooling_bwd_algo;

  113. 

  114. HandleImpl::HandleImpl(megcoreComputingHandle_t computing_handle, HandleType type)

  115.         : HandleImplHelper(computing_handle, type),

  116.           m_dispatcher{megcoreGetCPUDispatcher(computing_handle)} {}

  117. 

  118. size_t HandleImpl::image2d_pitch_alignment() const {

  119.     return g_image2d_pitch_alignment;

  120. }

  121. 

  122. HandleImpl::HandleVendorType HandleImpl::vendor_type() const {

  123.     return g_image2d_pitch_vendor;

  124. }

  125. 

  126. size_t HandleImpl::exchange_image2d_pitch_alignment(size_t alignment) {

  127.     auto ret = g_image2d_pitch_alignment;

  128.     g_image2d_pitch_alignment = alignment;

  129.     return ret;

  130. }

  131. 

  132. HandleImpl::HandleVendorType HandleImpl::exchange_image2d_vendor(

  133.         HandleImpl::HandleVendorType vendor) {

  134.     auto ret = g_image2d_pitch_vendor;

  135.     g_image2d_pitch_vendor = vendor;

  136.     return ret;

  137. }

  138. 

  139. MEGDNN_FOREACH_OPR_CLASS(MEGDNN_SPECIALIZE_CREATE_OPERATOR)

  140. 

  141. }  // namespace naive

  142. }  // namespace megdnn

  143. 

  144. // vim: syntax=cpp.doxygen