Skip to content

GitLab

Commit 9ee4967e authored by danyao12's avatar danyao12
Browse files

mi300 decoder test

parent 1ffad4d4
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 6 deletions
example/32_batched_gemm_scale_softmax_gemm/batched_multihead_attention_train_v1.cpp
View file @ 9ee4967e
...@@ -72,9 +72,9 @@ using Scale = ck::tensor_operation::element_wise::Scale; ...@@ -72,9 +72,9 @@ using Scale = ck::tensor_operation::element_wise::Scale;
using QKVElementOp = PassThrough; using QKVElementOp = PassThrough;
using YElementOp = PassThrough; using YElementOp = PassThrough;
using InputDataType = F16; using InputDataType = BF16;
using OutputDataType = F16; using OutputDataType = BF16;
using GemmDataType = F16; using GemmDataType = BF16;
using AccDataType = F32; using AccDataType = F32;
using ShuffleDataType = F32; using ShuffleDataType = F32;
using LSEDataType = F32; using LSEDataType = F32;
......
example/32_batched_gemm_scale_softmax_gemm/batched_multihead_attention_train_v2.cpp
View file @ 9ee4967e
...@@ -72,9 +72,9 @@ using Scale = ck::tensor_operation::element_wise::Scale; ...@@ -72,9 +72,9 @@ using Scale = ck::tensor_operation::element_wise::Scale;
using QKVElementOp = PassThrough; using QKVElementOp = PassThrough;
using YElementOp = PassThrough; using YElementOp = PassThrough;
using InputDataType = F16; using InputDataType = BF16;
using OutputDataType = F16; using OutputDataType = BF16;
using GemmDataType = F16; using GemmDataType = BF16;
using AccDataType = F32; using AccDataType = F32;
using ShuffleDataType = F32; using ShuffleDataType = F32;
using LSEDataType = F32; using LSEDataType = F32;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment