mirror of
https://github.com/Tencent/ncnn.git
synced 2026-10-17 04:41:47 +08:00
nihui
114 lines
3.1 KiB
Python
114 lines
3.1 KiB
Python
# Copyright 2021 Tencent
|
|
# SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
import pytest
|
|
|
|
import ncnn
|
|
|
|
|
|
def test_vk_blob_allocator():
|
|
if not hasattr(ncnn, "get_gpu_count"):
|
|
return
|
|
|
|
vkdev = ncnn.get_gpu_device(0)
|
|
assert vkdev is not None
|
|
allocator = ncnn.VkBlobAllocator(vkdev)
|
|
assert allocator.buffer_memory_type_index >= 0
|
|
assert allocator.image_memory_type_index >= 0
|
|
|
|
mappable = allocator.mappable
|
|
allocator.mappable = not mappable
|
|
assert allocator.mappable == (not mappable)
|
|
|
|
coherent = allocator.coherent
|
|
allocator.coherent = not coherent
|
|
assert allocator.coherent == (not coherent)
|
|
|
|
bufmem = allocator.fastMalloc(10 * 1024)
|
|
assert bufmem is not None
|
|
allocator.fastFree(bufmem)
|
|
|
|
imgmem = allocator.fastMalloc(4, 4, 3, 4, 1)
|
|
assert imgmem is not None
|
|
allocator.fastFree(imgmem)
|
|
|
|
|
|
def test_vk_weight_allocator():
|
|
if not hasattr(ncnn, "get_gpu_count"):
|
|
return
|
|
|
|
vkdev = ncnn.get_gpu_device(0)
|
|
assert vkdev is not None
|
|
allocator = ncnn.VkWeightAllocator(vkdev)
|
|
assert allocator.buffer_memory_type_index >= 0
|
|
assert allocator.image_memory_type_index >= 0
|
|
|
|
mappable = allocator.mappable
|
|
allocator.mappable = not mappable
|
|
assert allocator.mappable == (not mappable)
|
|
|
|
coherent = allocator.coherent
|
|
allocator.coherent = not coherent
|
|
assert allocator.coherent == (not coherent)
|
|
|
|
bufmem = allocator.fastMalloc(10 * 1024)
|
|
assert bufmem is not None
|
|
allocator.fastFree(bufmem)
|
|
|
|
imgmem = allocator.fastMalloc(4, 4, 3, 4, 1)
|
|
assert imgmem is not None
|
|
allocator.fastFree(imgmem)
|
|
|
|
|
|
def test_vk_staging_allocator():
|
|
if not hasattr(ncnn, "get_gpu_count"):
|
|
return
|
|
|
|
vkdev = ncnn.get_gpu_device(0)
|
|
assert vkdev is not None
|
|
allocator = ncnn.VkStagingAllocator(vkdev)
|
|
assert allocator.buffer_memory_type_index >= 0
|
|
assert allocator.image_memory_type_index >= 0
|
|
|
|
mappable = allocator.mappable
|
|
allocator.mappable = not mappable
|
|
assert allocator.mappable == (not mappable)
|
|
|
|
coherent = allocator.coherent
|
|
allocator.coherent = not coherent
|
|
assert allocator.coherent == (not coherent)
|
|
|
|
bufmem = allocator.fastMalloc(10 * 1024)
|
|
assert bufmem is not None
|
|
allocator.fastFree(bufmem)
|
|
|
|
imgmem = allocator.fastMalloc(4, 4, 3, 4, 1)
|
|
assert imgmem is not None
|
|
allocator.fastFree(imgmem)
|
|
|
|
|
|
def test_vk_weight_staging_allocator():
|
|
if not hasattr(ncnn, "get_gpu_count"):
|
|
return
|
|
|
|
vkdev = ncnn.get_gpu_device(0)
|
|
assert vkdev is not None
|
|
allocator = ncnn.VkWeightStagingAllocator(vkdev)
|
|
assert allocator.buffer_memory_type_index >= 0
|
|
assert allocator.image_memory_type_index >= 0
|
|
|
|
mappable = allocator.mappable
|
|
allocator.mappable = not mappable
|
|
assert allocator.mappable == (not mappable)
|
|
|
|
coherent = allocator.coherent
|
|
allocator.coherent = not coherent
|
|
assert allocator.coherent == (not coherent)
|
|
|
|
bufmem = allocator.fastMalloc(10 * 1024)
|
|
assert bufmem is not None
|
|
allocator.fastFree(bufmem)
|
|
|
|
imgmem = allocator.fastMalloc(4, 4, 3, 4, 1)
|
|
assert imgmem is None
|