diff options
Diffstat (limited to 'drivers/gpu/nvgpu/gk20a/as_gk20a.c')
-rw-r--r-- | drivers/gpu/nvgpu/gk20a/as_gk20a.c | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/drivers/gpu/nvgpu/gk20a/as_gk20a.c b/drivers/gpu/nvgpu/gk20a/as_gk20a.c index 7dcb8ed7..5acc626b 100644 --- a/drivers/gpu/nvgpu/gk20a/as_gk20a.c +++ b/drivers/gpu/nvgpu/gk20a/as_gk20a.c | |||
@@ -22,6 +22,8 @@ | |||
22 | 22 | ||
23 | #include <uapi/linux/nvgpu.h> | 23 | #include <uapi/linux/nvgpu.h> |
24 | 24 | ||
25 | #include <nvgpu/kmem.h> | ||
26 | |||
25 | #include "gk20a.h" | 27 | #include "gk20a.h" |
26 | 28 | ||
27 | /* dumb allocator... */ | 29 | /* dumb allocator... */ |
@@ -51,7 +53,7 @@ int gk20a_as_alloc_share(struct gk20a_as *as, | |||
51 | return -ENODEV; | 53 | return -ENODEV; |
52 | 54 | ||
53 | *out = NULL; | 55 | *out = NULL; |
54 | as_share = kzalloc(sizeof(*as_share), GFP_KERNEL); | 56 | as_share = nvgpu_kzalloc(g, sizeof(*as_share)); |
55 | if (!as_share) | 57 | if (!as_share) |
56 | return -ENOMEM; | 58 | return -ENOMEM; |
57 | 59 | ||
@@ -72,7 +74,7 @@ int gk20a_as_alloc_share(struct gk20a_as *as, | |||
72 | return 0; | 74 | return 0; |
73 | 75 | ||
74 | failed: | 76 | failed: |
75 | kfree(as_share); | 77 | nvgpu_kfree(g, as_share); |
76 | return err; | 78 | return err; |
77 | } | 79 | } |
78 | 80 | ||
@@ -98,8 +100,8 @@ int gk20a_as_release_share(struct gk20a_as_share *as_share) | |||
98 | 100 | ||
99 | release_fail: | 101 | release_fail: |
100 | release_as_share_id(as_share->as, as_share->id); | 102 | release_as_share_id(as_share->as, as_share->id); |
103 | nvgpu_kfree(g, as_share); | ||
101 | gk20a_put(g); | 104 | gk20a_put(g); |
102 | kfree(as_share); | ||
103 | 105 | ||
104 | return err; | 106 | return err; |
105 | } | 107 | } |