Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
e5e03c2c
"vscode:/vscode.git/clone" did not exist on "aa1e77a19ce658abcbaa0836f96878a7ae9dea84"
Unverified
Commit
e5e03c2c
authored
Mar 07, 2025
by
Lucas Wilkinson
Committed by
GitHub
Mar 06, 2025
Browse files
[BugFix] Illegal Memory Access in the blockwise cutlass fp8 GEMMs (#14396)
parent
e1744502
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
csrc/cutlass_extensions/gemm/collective/sm90_mma_tma_gmma_ss_warpspecialized_fp8_blockwise_scaling.hpp
...mma_tma_gmma_ss_warpspecialized_fp8_blockwise_scaling.hpp
+1
-1
No files found.
csrc/cutlass_extensions/gemm/collective/sm90_mma_tma_gmma_ss_warpspecialized_fp8_blockwise_scaling.hpp
View file @
e5e03c2c
...
@@ -402,7 +402,7 @@ struct CollectiveMma<
...
@@ -402,7 +402,7 @@ struct CollectiveMma<
// TODO: test `scale_copy_a` with `ScaleMsPerTile` < 128
// TODO: test `scale_copy_a` with `ScaleMsPerTile` < 128
TiledCopy
scale_copy_a
=
make_tiled_copy
(
SmemBlockScalingCopyAtomA
{},
TiledCopy
scale_copy_a
=
make_tiled_copy
(
SmemBlockScalingCopyAtomA
{},
Layout
<
Shape
<
_32
,
_1
>>
{},
Layout
<
Shape
<
_4
,
_1
>>
{});
// (1,1,1)
Layout
<
Shape
<
_32
>>
{},
Layout
<
Shape
<
_1
>>
{});
// (1,1,1)
TiledCopy
scale_copy_b
=
make_tiled_copy
(
SmemBlockScalingCopyAtomB
{},
TiledCopy
scale_copy_b
=
make_tiled_copy
(
SmemBlockScalingCopyAtomB
{},
Layout
<
Shape
<
_1
>>
{},
Layout
<
Shape
<
_1
>>
{});
// (1,1,1)
Layout
<
Shape
<
_1
>>
{},
Layout
<
Shape
<
_1
>>
{});
// (1,1,1)
ThrCopy
thr_scale_copy_a
=
scale_copy_a
.
get_slice
(
threadIdx
.
x
);
ThrCopy
thr_scale_copy_a
=
scale_copy_a
.
get_slice
(
threadIdx
.
x
);
...
...
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