mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-04 21:51:28 +00:00
Fix and optimize dynamic unary elementwise (#1818)
* Fix and optimize dynamic unary elementwise * fix
This commit is contained in:
@@ -1,5 +1,5 @@
|
||||
// SPDX-License-Identifier: MIT
|
||||
// Copyright (c) 2023-2024, Advanced Micro Devices, Inc. All rights reserved.
|
||||
// Copyright (c) 2023-2025, Advanced Micro Devices, Inc. All rights reserved.
|
||||
|
||||
#pragma once
|
||||
|
||||
@@ -121,19 +121,6 @@ __global__ void
|
||||
static_for<0, NumDTensor, 1>{}(
|
||||
[&](auto i) { p_ds_grid_grp(i) = p_ds_grid[i] + ds_group_offset[i]; });
|
||||
|
||||
if constexpr(is_same_v<AElementwiseOperation, element_wise::DynamicUnaryOp>)
|
||||
{
|
||||
a_element_op.InitUnaryOpPtrOnDevice();
|
||||
}
|
||||
if constexpr(is_same_v<BElementwiseOperation, element_wise::DynamicUnaryOp>)
|
||||
{
|
||||
b_element_op.InitUnaryOpPtrOnDevice();
|
||||
}
|
||||
if constexpr(is_same_v<CDEElementwiseOperation, element_wise::DynamicUnaryOp>)
|
||||
{
|
||||
cde_element_op.InitUnaryOpPtrOnDevice();
|
||||
}
|
||||
|
||||
if constexpr(isMultiA || isMultiB)
|
||||
{
|
||||
AsPointer p_as_grid_grp;
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
Reference in New Issue
Block a user