提交 103d7f33 编写于 作者: M Megvii Engine Team

refactor(dnn/rocm): update hip license header

GitOrigin-RevId: 79d684755d31a982446a2c136ce800d70afae91f
上级 5aa52d38
/**
* \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 "./add_update.h.hip"
......
/**
*
* \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
......
/**
* \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 "hip_header.h"
......
......@@ -37,7 +37,7 @@ void ArgsortForwardImpl::exec(_megdnn_tensor_in src, _megdnn_tensor_out dst,
ARGSORT_FOREACH_CTYPE(cb);
#undef cb
default:
megdnn_throw(ssprintf("unsupported argsort dtype on cuda: %s",
megdnn_throw(ssprintf("unsupported argsort dtype on rocm: %s",
src.layout.dtype.name()));
}
}
......@@ -71,7 +71,7 @@ void ArgsortBackwardImpl::exec(_megdnn_tensor_in diff,
ARGSORT_FOREACH_CTYPE(cb);
#undef cb
default:
megdnn_throw(ssprintf("unsupported argsort dtype on cuda: %s",
megdnn_throw(ssprintf("unsupported argsort dtype on rocm: %s",
diff.layout.dtype.name()));
}
}
......
/**
* \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 "hip_header.h"
......
/**
* \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
......
/**
* \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"
......
/**
* \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"
......
/**
* \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"
......
/**
* \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
......
/**
* \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
......
/**
* \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 "src/rocm/utils.h.hip"
......
/**
* \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
......
/**
* \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 "megdnn/dtype.h"
......
/**
* \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
......
/**
* \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
......
/**
* \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
* developed by Megvii.
......
/**
* \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
......
/**
* \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
......
/**
* \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 "hip_header.h"
......
/**
* \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
#include <stdint.h>
......
/**
* \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
......
/**
* \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"
......
/**
* \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
......
/**
* \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"
......
/**
* \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"
......
/**
* \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"
......
/**
* \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 "./indexing_one_hot.h.hip"
......
/**
* \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
......
/**
* \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
......
/**
* \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 "./linspace.h.hip"
......
/**
* \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"
......
/**
* \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 "src/rocm/powc/powc.h.hip"
......
/**
* \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"
......
/**
* \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 "hip_header.h"
......
/**
* \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
......
/**
* \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
......
/**
* \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
......
/**
* \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
......
/**
* \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 "./relayout.h.hip"
......
/**
* \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
......
/**
* \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"
......
/**
* \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
......
/**
* \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 "hip_header.h"
......
/**
* \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
......
......@@ -81,18 +81,10 @@ void TopKImpl::do_exec(int k, _megdnn_tensor_in data, _megdnn_tensor_out values,
values.ptr<int32_t>(), indices,
workspace.raw_ptr);
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:
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()));
}
}
......
......@@ -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
/*!
......
/**
* \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 "./type_cvt.h.hip"
......
/**
* \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"
......
/**
* \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
......
......@@ -29,10 +29,5 @@ TEST_F(ROCM, TOP_K) {
TEST_F(ROCM, TOP_K_I32) {
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
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册