| @@ -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 | ||||