已合并
Remove Ascend950 version restriction for ACLNN support in _chunk_cat function #5357
nomiz创建于 7月2日
Remove Ascend950 version restriction for ACLNN support in _chunk_cat function #5357
已合并
共 1 个文件变更+2-4
| @@ -88,8 +88,7 @@ namespace op_api { | |||
| 88 | 88 | ||
| 89 | at::Tensor _chunk_cat( at::TensorList tensors, int64_t dim, int64_t num_chunks) | 89 | at::Tensor _chunk_cat( at::TensorList tensors, int64_t dim, int64_t num_chunks) |
| 90 | { | 90 | { |
| 91 | - static bool npu_support_aclnn = check_aclnn_kernel_available("aclnnChunkCat") | 91 | + static bool npu_support_aclnn = check_aclnn_kernel_available("aclnnChunkCat"); |
| 92 | - && c10_npu::GetSocVersion() < c10_npu::SocVersion::Ascend950; | ||
| 93 | if (!npu_support_aclnn) { | 92 | if (!npu_support_aclnn) { |
| 94 | return at::native::_chunk_cat(tensors, dim, num_chunks); | 93 | return at::native::_chunk_cat(tensors, dim, num_chunks); |
| 95 | } | 94 | } |
| @@ -108,8 +107,7 @@ namespace op_api { | |||
| 108 | 107 | ||
| 109 | at::Tensor& _chunk_cat_out( at::TensorList tensors, int64_t dim, int64_t num_chunks, at::Tensor& out) | 108 | at::Tensor& _chunk_cat_out( at::TensorList tensors, int64_t dim, int64_t num_chunks, at::Tensor& out) |
| 110 | { | 109 | { |
| 111 | - static bool npu_support_aclnn = check_aclnn_kernel_available("aclnnChunkCat") | 110 | + static bool npu_support_aclnn = check_aclnn_kernel_available("aclnnChunkCat"); |
| 112 | - && c10_npu::GetSocVersion() < c10_npu::SocVersion::Ascend950; | ||
| 113 | if (!npu_support_aclnn) { | 111 | if (!npu_support_aclnn) { |
| 114 | return at::native::_chunk_cat_out(tensors, dim, num_chunks, out); | 112 | return at::native::_chunk_cat_out(tensors, dim, num_chunks, out); |
| 115 | } | 113 | } |