Buffer overflow at CudaAttentionOpTest::selfAttentionOpTest

[skmkt]

Definition: std::vector<void*> block_pointers(batch_size * 2 * maxBlocksPerSeq, nullptr);

... auto kv_cache = device_->allocateBuffer( {DataType::TYPE_UINT64, {(size_t)batch_size, maxBlocksPerSeq}, AllocationType::HOST}, {});

Copy size larger than dst size: std::memcpy(kv_cache->data(), block_pointers.data(), block_pointers.size() * sizeof(void*));

block_pointers.size() * sizeof(void*) = batch_size * 2 * maxBlocksPerSeq * sizeof(void*) kv_cache size() = batch_size * maxBlocksPerSeq * sizeof(unsigned long)

[netaddi]

Hi, thank you for your comment on this bug. we are working on a fix now. btw, how did you find this ?