59 lines · c
1// SPDX-License-Identifier: GPL-2.02 3/*4 * Copyright 2016-2019 HabanaLabs, Ltd.5 * All Rights Reserved.6 */7 8#include "habanalabs.h"9 10#include <linux/slab.h>11 12int hl_asid_init(struct hl_device *hdev)13{14 hdev->asid_bitmap = bitmap_zalloc(hdev->asic_prop.max_asid, GFP_KERNEL);15 if (!hdev->asid_bitmap)16 return -ENOMEM;17 18 mutex_init(&hdev->asid_mutex);19 20 /* ASID 0 is reserved for the kernel driver and device CPU */21 set_bit(0, hdev->asid_bitmap);22 23 return 0;24}25 26void hl_asid_fini(struct hl_device *hdev)27{28 mutex_destroy(&hdev->asid_mutex);29 bitmap_free(hdev->asid_bitmap);30}31 32unsigned long hl_asid_alloc(struct hl_device *hdev)33{34 unsigned long found;35 36 mutex_lock(&hdev->asid_mutex);37 38 found = find_first_zero_bit(hdev->asid_bitmap,39 hdev->asic_prop.max_asid);40 if (found == hdev->asic_prop.max_asid)41 found = 0;42 else43 set_bit(found, hdev->asid_bitmap);44 45 mutex_unlock(&hdev->asid_mutex);46 47 return found;48}49 50void hl_asid_free(struct hl_device *hdev, unsigned long asid)51{52 if (asid == HL_KERNEL_ASID_ID || asid >= hdev->asic_prop.max_asid) {53 dev_crit(hdev->dev, "Invalid ASID %lu", asid);54 return;55 }56 57 clear_bit(asid, hdev->asid_bitmap);58}59