提交 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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* *
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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, ...@@ -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, ...@@ -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()));
} }
} }
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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
......
/** /**
* \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.
......
/** /**
* \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
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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>
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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, ...@@ -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 " ssprintf("only float32, int32 are supported for "
"cuda topk, got: %s", "rocm topk, got: %s",
data.layout.dtype.name())); data.layout.dtype.name()));
} }
} }
......
...@@ -59,29 +59,6 @@ struct RadixConverter<int32_t> { ...@@ -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
/*! /*!
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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"
......
/** /**
* \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 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
* developed by Megvii.
* *
* \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) { ...@@ -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
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册