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
FlashMLA
Commits
ae382f02
Commit
ae382f02
authored
Feb 27, 2026
by
zhanghj2
Browse files
assert FLASH_MLA_ASM_DIR
parent
c353b35b
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
csrc/extension/flash_fwd_mla_kernel_fp8.h
csrc/extension/flash_fwd_mla_kernel_fp8.h
+1
-1
No files found.
csrc/extension/flash_fwd_mla_kernel_fp8.h
View file @
ae382f02
...
@@ -2524,7 +2524,7 @@ void run_flash_splitkv_fwd_mla_fp8_tp1(Flash_fwd_mla_params ¶ms, cudaStream_
...
@@ -2524,7 +2524,7 @@ void run_flash_splitkv_fwd_mla_fp8_tp1(Flash_fwd_mla_params ¶ms, cudaStream_
}
}
else
{
else
{
static
char
*
FLASH_MLA_ASM_DIR
=
std
::
getenv
(
"FLASH_MLA_ROOT_DIR"
);
static
char
*
FLASH_MLA_ASM_DIR
=
std
::
getenv
(
"FLASH_MLA_ROOT_DIR"
);
static_
assert
(
FLASH_MLA_ASM_DIR
);
assert
(
FLASH_MLA_ASM_DIR
!=
nullptr
&&
"FLASH_MLA_ASM_DIR nullptr
\n
"
);
constexpr
size_t
smem_size
=
65536
;
constexpr
size_t
smem_size
=
65536
;
std
::
string
co_file
=
std
::
string
(
FLASH_MLA_ASM_DIR
)
+
std
::
string
co_file
=
std
::
string
(
FLASH_MLA_ASM_DIR
)
+
"flash_fwd_mla_fp8_gfx938-hip-amdgcn-amd-amdhsa-gfx938.co"
;
"flash_fwd_mla_fp8_gfx938-hip-amdgcn-amd-amdhsa-gfx938.co"
;
...
...
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