Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
composable_kernel_ROCM
Commits
85a7aa89
"driver/vscode:/vscode.git/clone" did not exist on "f2f35201f40d2a36c6c475554ec2b33fc4e0bbf1"
Commit
85a7aa89
authored
Feb 13, 2025
by
mtgu0705
Browse files
fix bug, function now passes.
parent
279e2eaf
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
9 additions
and
7 deletions
+9
-7
include/ck/tensor_operation/gpu/device/impl/device_gemm_xdl_cshuffle_v3_b_preshuffle.hpp
.../device/impl/device_gemm_xdl_cshuffle_v3_b_preshuffle.hpp
+6
-4
include/ck/tensor_operation/gpu/thread/threadwise_tensor_slice_transfer.hpp
...operation/gpu/thread/threadwise_tensor_slice_transfer.hpp
+3
-3
No files found.
include/ck/tensor_operation/gpu/device/impl/device_gemm_xdl_cshuffle_v3_b_preshuffle.hpp
View file @
85a7aa89
...
@@ -339,19 +339,21 @@ struct DeviceGemm_Xdl_CShuffleV3_BPreshuffle : public DeviceGemmV2BPreshuffle<AL
...
@@ -339,19 +339,21 @@ struct DeviceGemm_Xdl_CShuffleV3_BPreshuffle : public DeviceGemmV2BPreshuffle<AL
if(arg.KBatch > 1)
if(arg.KBatch > 1)
{
{
const auto kernel =
const auto kernel =
kernel_gemm_xdl_cshuffle_v3<GridwiseGemm,
kernel_gemm_xdl_cshuffle_v3
_b_preshuffle
<GridwiseGemm,
false,
false,
InMemoryDataOperationEnum::AtomicAdd,
InMemoryDataOperationEnum::AtomicAdd,
minimum_occupancy>;
minimum_occupancy,
TailNumber::Odd>;
Run(kernel);
Run(kernel);
}
}
else
else
{
{
const auto kernel =
const auto kernel =
kernel_gemm_xdl_cshuffle_v3<GridwiseGemm,
kernel_gemm_xdl_cshuffle_v3
_b_preshuffle
<GridwiseGemm,
false,
false,
InMemoryDataOperationEnum::Set,
InMemoryDataOperationEnum::Set,
minimum_occupancy>;
minimum_occupancy,
TailNumber::Odd>;
Run(kernel);
Run(kernel);
}
}
}
}
...
...
include/ck/tensor_operation/gpu/thread/threadwise_tensor_slice_transfer.hpp
View file @
85a7aa89
...
@@ -377,14 +377,14 @@ struct ThreadwiseTensorSliceTransfer_v2
...
@@ -377,14 +377,14 @@ struct ThreadwiseTensorSliceTransfer_v2
if
constexpr
(
InvalidElementAsNaN
)
if
constexpr
(
InvalidElementAsNaN
)
{
{
dst_buf
(
Number
<
dst_offset
/
PackedSize
>
{})
=
dst_buf
(
Number
<
dst_offset
>
{})
=
is_src_valid
is_src_valid
?
type_convert
<
DstData
>
(
src_vector
.
template
AsType
<
SrcData
>()[
i
])
?
type_convert
<
DstData
>
(
src_vector
.
template
AsType
<
SrcData
>()[
i
])
:
NumericLimits
<
DstData
>::
QuietNaN
();
:
NumericLimits
<
DstData
>::
QuietNaN
();
}
}
else
else
{
{
dst_buf
(
Number
<
dst_offset
/
PackedSize
>
{})
=
dst_buf
(
Number
<
dst_offset
>
{})
=
type_convert
<
DstData
>
(
src_vector
.
template
AsType
<
SrcData
>()[
i
]);
type_convert
<
DstData
>
(
src_vector
.
template
AsType
<
SrcData
>()[
i
]);
}
}
});
});
...
@@ -1619,7 +1619,7 @@ struct ThreadwiseTensorSliceTransfer_StaticToStatic
...
@@ -1619,7 +1619,7 @@ struct ThreadwiseTensorSliceTransfer_StaticToStatic
constexpr
index_t
src_offset
=
src_desc
.
CalculateOffset
(
constexpr
index_t
src_offset
=
src_desc
.
CalculateOffset
(
src_slice_origin_idx
+
idx_md
+
i
*
dst_scalar_step_in_vector
);
src_slice_origin_idx
+
idx_md
+
i
*
dst_scalar_step_in_vector
);
src_tmp_vector
.
template
AsType
<
SrcData
>()(
i
)
=
src_buf
[
Number
<
src_offset
/
PackedSize
>
{}];
src_tmp_vector
.
template
AsType
<
SrcData
>()(
i
)
=
src_buf
[
Number
<
src_offset
>
{}];
});
});
// copy data from src_tmp_vector to dst_tmp_vector (data cast data from SrcData to
// copy data from src_tmp_vector to dst_tmp_vector (data cast data from SrcData to
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment