add new system file

hc

2024-02-20 102a0743326a03cd1a1202ceda21e175b7d3575c

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