@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/add_update/add_update.cpp.hip | |||||
* \file dnn/src/rocm/add_update/add_update.cpp.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hcc_detail/hcc_defs_prologue.h" | #include "hcc_detail/hcc_defs_prologue.h" | ||||
#include "./add_update.h.hip" | #include "./add_update.h.hip" | ||||
@@ -1,11 +1,14 @@ | |||||
/** | /** | ||||
* | * | ||||
* \file src/rocm/add_update/add_update.h.hip | |||||
* \file dnn/src/rocm/add_update/add_update.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/argmxx/argmxx.cpp.hip | |||||
* \file dnn/src/rocm/argmxx/argmxx.cpp.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hcc_detail/hcc_defs_prologue.h" | #include "hcc_detail/hcc_defs_prologue.h" | ||||
#include "hip_header.h" | #include "hip_header.h" | ||||
@@ -37,7 +37,7 @@ void ArgsortForwardImpl::exec(_megdnn_tensor_in src, _megdnn_tensor_out dst, | |||||
ARGSORT_FOREACH_CTYPE(cb); | ARGSORT_FOREACH_CTYPE(cb); | ||||
#undef cb | #undef cb | ||||
default: | default: | ||||
megdnn_throw(ssprintf("unsupported argsort dtype on cuda: %s", | |||||
megdnn_throw(ssprintf("unsupported argsort dtype on rocm: %s", | |||||
src.layout.dtype.name())); | src.layout.dtype.name())); | ||||
} | } | ||||
} | } | ||||
@@ -71,7 +71,7 @@ void ArgsortBackwardImpl::exec(_megdnn_tensor_in diff, | |||||
ARGSORT_FOREACH_CTYPE(cb); | ARGSORT_FOREACH_CTYPE(cb); | ||||
#undef cb | #undef cb | ||||
default: | default: | ||||
megdnn_throw(ssprintf("unsupported argsort dtype on cuda: %s", | |||||
megdnn_throw(ssprintf("unsupported argsort dtype on rocm: %s", | |||||
diff.layout.dtype.name())); | diff.layout.dtype.name())); | ||||
} | } | ||||
} | } | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/checksum/kern.cpp.hip | |||||
* \file dnn/src/rocm/checksum/kern.cpp.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hcc_detail/hcc_defs_prologue.h" | #include "hcc_detail/hcc_defs_prologue.h" | ||||
#include "hip_header.h" | #include "hip_header.h" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/checksum/kern.h.hip | |||||
* \file dnn/src/rocm/checksum/kern.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/convolution/chanwise/bwd_data.cpp.hip | |||||
* \file dnn/src/rocm/convolution/chanwise/bwd_data.cpp.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hip_header.h" | #include "hip_header.h" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/convolution/chanwise/bwd_filter.cpp.hip | |||||
* \file dnn/src/rocm/convolution/chanwise/bwd_filter.cpp.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hip_header.h" | #include "hip_header.h" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/convolution/chanwise/fwd.cpp.hip | |||||
* \file dnn/src/rocm/convolution/chanwise/fwd.cpp.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hip_header.h" | #include "hip_header.h" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/convolution/chanwise/kern.h.hip | |||||
* \file dnn/src/rocm/convolution/chanwise/kern.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/convolution/chanwise/kern_helper.h.hip | |||||
* \file dnn/src/rocm/convolution/chanwise/kern_helper.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/convolution/forward/inplace_matmul_impl.cpp.hip | |||||
* \file dnn/src/rocm/convolution/forward/inplace_matmul_impl.cpp.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "./inplace_matmul_impl.h.hip" | #include "./inplace_matmul_impl.h.hip" | ||||
#include "src/rocm/utils.h.hip" | #include "src/rocm/utils.h.hip" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/convolution/forward/inplace_matmul_impl.h.hip | |||||
* \file dnn/src/rocm/convolution/forward/inplace_matmul_impl.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/convolution/im2col.cpp.hip | |||||
* \file dnn/src/rocm/convolution/im2col.cpp.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "./im2col.h.hip" | #include "./im2col.h.hip" | ||||
#include "megdnn/dtype.h" | #include "megdnn/dtype.h" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/convolution/im2col.h.hip | |||||
* \file dnn/src/rocm/convolution/im2col.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
@@ -1,12 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/elemwise/kern_wrapper.h.hip | |||||
* \file dnn/src/rocm/elemwise/kern_wrapper.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \brief helper for implementing elemwise oprs | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
@@ -1,5 +1,5 @@ | |||||
/** | /** | ||||
* \file src/rocm/elemwise/special_kerns.h.hip | |||||
* \file dnn/src/rocm/elemwise/special_kerns.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | * This file is part of MegDNN, a deep neural network run-time library | ||||
* developed by Megvii. | * developed by Megvii. | ||||
@@ -1,11 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/elemwise_helper.h.hip | |||||
* \file dnn/src/rocm/elemwise_helper.h.hip | |||||
* | * | ||||
* This file is part of MegBrain, a deep learning framework developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \brief helper utilities for implementing element-wise kernels | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/error_info.h.hip | |||||
* \file dnn/src/rocm/error_info.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/eye/eye.cpp.hip | |||||
* \file dnn/src/rocm/eye/eye.cpp.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hcc_detail/hcc_defs_prologue.h" | #include "hcc_detail/hcc_defs_prologue.h" | ||||
#include "hip_header.h" | #include "hip_header.h" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/eye/eye.h.hip | |||||
* \file dnn/src/rocm/eye/eye.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
#include <stdint.h> | #include <stdint.h> | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/indexing_multi_axis_vec/kern.h.hip | |||||
* \file dnn/src/rocm/indexing_multi_axis_vec/kern.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/indexing_multi_axis_vec/kern_apply_opr_fwd.cpp.hip | |||||
* \file dnn/src/rocm/indexing_multi_axis_vec/kern_apply_opr_fwd.cpp.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hcc_detail/hcc_defs_prologue.h" | #include "hcc_detail/hcc_defs_prologue.h" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/indexing_multi_axis_vec/kern_apply_opr_impl.hipinl | |||||
* \file dnn/src/rocm/indexing_multi_axis_vec/kern_apply_opr_impl.hipinl | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#ifndef KERN_APPLY_OPR_OPR | #ifndef KERN_APPLY_OPR_OPR | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/indexing_multi_axis_vec/kern_apply_opr_incr.cpp.hip | |||||
* \file dnn/src/rocm/indexing_multi_axis_vec/kern_apply_opr_incr.cpp.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hcc_detail/hcc_defs_prologue.h" | #include "hcc_detail/hcc_defs_prologue.h" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/indexing_multi_axis_vec/kern_apply_opr_set.cpp.hip | |||||
* \file dnn/src/rocm/indexing_multi_axis_vec/kern_apply_opr_set.cpp.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hcc_detail/hcc_defs_prologue.h" | #include "hcc_detail/hcc_defs_prologue.h" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/indexing_multi_axis_vec/kern_gen_offset_base.hip.cpp | |||||
* \file dnn/src/rocm/indexing_multi_axis_vec/kern_gen_offset_base.hip.cpp | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hcc_detail/hcc_defs_prologue.h" | #include "hcc_detail/hcc_defs_prologue.h" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/indexing_one_hot/indexing_one_hot.cpp.hip | |||||
* \file dnn/src/rocm/indexing_one_hot/indexing_one_hot.cpp.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2016 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hcc_detail/hcc_defs_prologue.h" | #include "hcc_detail/hcc_defs_prologue.h" | ||||
#include "./indexing_one_hot.h.hip" | #include "./indexing_one_hot.h.hip" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/indexing_one_hot/indexing_one_hot.h.hip | |||||
* \file dnn/src/rocm/indexing_one_hot/indexing_one_hot.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
@@ -1,12 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/int_fastdiv.h.hip | |||||
* \file dnn/src/rocm/int_fastdiv.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \brief fast integer division for constant divisor | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/linspace/linspace.cpp.hip | |||||
* \file dnn/src/rocm/linspace/linspace.cpp.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hcc_detail/hcc_defs_prologue.h" | #include "hcc_detail/hcc_defs_prologue.h" | ||||
#include "./linspace.h.hip" | #include "./linspace.h.hip" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/linspace/linspace.h.hip | |||||
* \file dnn/src/rocm/linspace/linspace.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hip_header.h" | #include "hip_header.h" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/powc/powc.cpp.hip | |||||
* \file dnn/src/rocm/powc/powc.cpp.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hcc_detail/hcc_defs_prologue.h" | #include "hcc_detail/hcc_defs_prologue.h" | ||||
#include "src/rocm/powc/powc.h.hip" | #include "src/rocm/powc/powc.h.hip" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/powc/powc.h.hip | |||||
* \file dnn/src/rocm/powc/powc.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hip_header.h" | #include "hip_header.h" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/reduce/reduce.cpp.hip | |||||
* \file dnn/src/rocm/reduce/reduce.cpp.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hcc_detail/hcc_defs_prologue.h" | #include "hcc_detail/hcc_defs_prologue.h" | ||||
#include "hip_header.h" | #include "hip_header.h" | ||||
@@ -1,11 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/reduce_helper.h.hip | |||||
* \file dnn/src/rocm/reduce_helper.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \brief helper for implementing reduce operators | |||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
@@ -1,12 +1,14 @@ | |||||
/** | /** | ||||
* \file src/rocm/reduce_helper.hipinl | |||||
* \file dnn/src/rocm/reduce_helper.hipinl | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \brief helper for implementing reduce operators | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
@@ -1,12 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/reduce_helper/column.hipinl | |||||
* \file dnn/src/rocm/reduce_helper/column.hipinl | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \brief reduce kernel for (A, B) -> (A, 1) with small B | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
@@ -1,12 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/reduce_helper/largeBC.hipinl | |||||
* \file dnn/src/rocm/reduce_helper/largeBC.hipinl | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \brief reduce kernel for large (B * C) value | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/relayout/relayout.cpp.hip | |||||
* \file dnn/src/rocm/relayout/relayout.cpp.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hcc_detail/hcc_defs_prologue.h" | #include "hcc_detail/hcc_defs_prologue.h" | ||||
#include "./relayout.h.hip" | #include "./relayout.h.hip" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/relayout/relayout.h.hip | |||||
* \file dnn/src/rocm/relayout/relayout.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/relayout/kern_contiguous.cpp.hip | |||||
* \file dnn/src/rocm/relayout/kern_contiguous.cpp.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hcc_detail/hcc_defs_prologue.h" | #include "hcc_detail/hcc_defs_prologue.h" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/relayout/relayout_contiguous.h.hip | |||||
* \file dnn/src/rocm/relayout/relayout_contiguous.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/sleep/kern.cpp.hip | |||||
* \file dnn/src/rocm/sleep/kern.cpp.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hcc_detail/hcc_defs_prologue.h" | #include "hcc_detail/hcc_defs_prologue.h" | ||||
#include "hip_header.h" | #include "hip_header.h" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/sleep/kern.h.hip | |||||
* \file dnn/src/rocm/sleep/kern.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
@@ -81,18 +81,10 @@ void TopKImpl::do_exec(int k, _megdnn_tensor_in data, _megdnn_tensor_out values, | |||||
values.ptr<int32_t>(), indices, | values.ptr<int32_t>(), indices, | ||||
workspace.raw_ptr); | workspace.raw_ptr); | ||||
return; | return; | ||||
// #if !MEGDNN_DISABLE_FLOAT16 | |||||
// case DTypeEnum::Float16: | |||||
// dispatch_with_ctype<dt_float16>(k, data.layout[0], data.layout[1], | |||||
// data.layout.stride[0], data.ptr<dt_float16>(), | |||||
// values.ptr<dt_float16>(), indices, | |||||
// workspace.raw_ptr); | |||||
// return; | |||||
// #endif | |||||
default: | default: | ||||
megdnn_throw( | megdnn_throw( | ||||
ssprintf("only float32, int32 and float16 supported for " | |||||
"cuda topk, got: %s", | |||||
ssprintf("only float32, int32 are supported for " | |||||
"rocm topk, got: %s", | |||||
data.layout.dtype.name())); | data.layout.dtype.name())); | ||||
} | } | ||||
} | } | ||||
@@ -59,29 +59,6 @@ struct RadixConverter<int32_t> { | |||||
} | } | ||||
}; | }; | ||||
// #if !MEGDNN_DISABLE_FLOAT16 | |||||
// template <> | |||||
// struct RadixConverter<dt_float16> { | |||||
// union FIunion { | |||||
// FIunion() {} | |||||
// dt_float16 fv; | |||||
// uint16_t iv; | |||||
// }; | |||||
// static __forceinline__ __device__ __host__ uint16_t to_radix(dt_float16 val) { | |||||
// FIunion fi; | |||||
// fi.fv = val; | |||||
// return fi.iv ^ (((!(fi.iv >> 15u)) - 1u) | 0x8000u); | |||||
// } | |||||
// static __forceinline__ __device__ __host__ dt_float16 from_radix(uint16_t val) { | |||||
// FIunion fi; | |||||
// // do not write as to_radix() to work around a compiler bug in cuda-9.0 | |||||
// uint16_t m = 0x8000u; | |||||
// fi.iv = val ^ (m | (m - !(val >> 15u))); | |||||
// return fi.fv; | |||||
// } | |||||
// }; | |||||
// #endif | |||||
} // namespace internal | } // namespace internal | ||||
/*! | /*! | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/type_cvt/type_cvt.cpp.hip | |||||
* \file dnn/src/rocm/type_cvt/type_cvt.cpp.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hcc_detail/hcc_defs_prologue.h" | #include "hcc_detail/hcc_defs_prologue.h" | ||||
#include "./type_cvt.h.hip" | #include "./type_cvt.h.hip" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/type_cvt/type_cvt.h.hip | |||||
* \file dnn/src/rocm/type_cvt/type_cvt.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#include "hip_header.h" | #include "hip_header.h" | ||||
@@ -1,10 +1,13 @@ | |||||
/** | /** | ||||
* \file src/rocm/utils.h.hip | |||||
* \file dnn/src/rocm/utils.h.hip | |||||
* | * | ||||
* This file is part of MegDNN, a deep neural network run-time library | |||||
* developed by Megvii. | |||||
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") | |||||
* | * | ||||
* \copyright Copyright (c) 2014-2019 Megvii Inc. All rights reserved. | |||||
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved. | |||||
* | |||||
* Unless required by applicable law or agreed to in writing, | |||||
* software distributed under the License is distributed on an | |||||
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||||
*/ | */ | ||||
#pragma once | #pragma once | ||||
@@ -29,10 +29,5 @@ TEST_F(ROCM, TOP_K) { | |||||
TEST_F(ROCM, TOP_K_I32) { | TEST_F(ROCM, TOP_K_I32) { | ||||
run_topk_test<dtype::Int32>(handle_rocm()); | run_topk_test<dtype::Int32>(handle_rocm()); | ||||
} | } | ||||
// #if !MEGDNN_DISABLE_FLOAT16 | |||||
// TEST_F(ROCM, TOP_K_F16) { | |||||
// run_topk_test<dtype::Float16>(handle_rocm()); | |||||
// } | |||||
// #endif | |||||
// vim: syntax=cpp.doxygen | // vim: syntax=cpp.doxygen |