Skip to content

GitLab

"...unconditional_image_generation/train_unconditional.py" did not exist on "7fe05bb311bf2847c9918a85c9ba22ca6d2842b2"
Commit f9ffbc9c authored by Jing Zhang's avatar Jing Zhang Committed by Gu
Browse files

fixed assert

parent a33e0b10
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
include/ck/tensor_operation/gpu/thread/threadwise_tensor_slice_transfer_v3r1.hpp
View file @ f9ffbc9c
...@@ -375,8 +375,8 @@ struct ThreadwiseTensorSliceTransfer_v3r1 ...@@ -375,8 +375,8 @@ struct ThreadwiseTensorSliceTransfer_v3r1
(is_same<f8_t, remove_cvref_t<DstData>>::value && (is_same<f8_t, remove_cvref_t<DstData>>::value &&
SrcScalarPerVector % 4 == 0 && DstScalarPerVector % 4 == 0))) SrcScalarPerVector % 4 == 0 && DstScalarPerVector % 4 == 0)))
{ {
static_assert(is_same_v<remove_cvref_t<SrcData>, pk_i4_t>, static_assert(!is_same_v<remove_cvref_t<SrcData>, pk_i4_t>,
"transpose is not allowed for pk_i4_t"); "in-register transpose is not supported for pk_i4_t");
// each transpose does // each transpose does
// DstScalarPerVector # of src vectors in src_thread_scratch_ // DstScalarPerVector # of src vectors in src_thread_scratch_
// SrcScalarPerVector # of dst vectors in dst_thread_scratch_ // SrcScalarPerVector # of dst vectors in dst_thread_scratch_
......
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