@@ -145,6 +145,8 @@ VulkanBuffer* CreateBuffer(const VulkanContext& vctx, size_t nbytes, VkBufferUsa
145
145
146
146
uint32_t mem_type_index = vctx.compute_mtype_index ;
147
147
148
+ LOG (INFO) << " compute_mtype_index: " << vctx.compute_mtype_index ;
149
+
148
150
if (usage & VK_BUFFER_USAGE_UNIFORM_BUFFER_BIT) {
149
151
// Find a memory type that supports UBO
150
152
auto prop = VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT | VK_MEMORY_PROPERTY_HOST_COHERENT_BIT;
@@ -185,6 +187,7 @@ VulkanBuffer* CreateBuffer(const VulkanContext& vctx, size_t nbytes, VkBufferUsa
185
187
minfo.memoryTypeIndex = mem_type_index;
186
188
VULKAN_CALL (vkAllocateMemory (vctx.device , &minfo, nullptr , &memory));
187
189
} else {
190
+ LOG (INFO) << " using dedicated allocation" ;
188
191
VkMemoryAllocateInfo minfo;
189
192
minfo.sType = VK_STRUCTURE_TYPE_MEMORY_ALLOCATE_INFO;
190
193
minfo.pNext = nullptr ;
@@ -1174,6 +1177,7 @@ void VulkanWrappedFunc::operator()(TVMArgs args, TVMRetValue* rv,
1174
1177
descriptor_buffers.push_back (binfo);
1175
1178
}
1176
1179
if (vctx.UseImmediate ()) {
1180
+ LOG (INFO) << " Using immediate" ;
1177
1181
// Can safely capture by reference as this lambda is immediately executed on the calling thread.
1178
1182
VulkanThreadEntry::ThreadLocal ()->Stream (device_id)->Launch ([&](VulkanStreamState* state) {
1179
1183
vkCmdBindPipeline (state->cmd_buffer_ , VK_PIPELINE_BIND_POINT_COMPUTE, pipeline->pipeline );
0 commit comments