1/* 2 * SPU file system 3 * 4 * (C) Copyright IBM Deutschland Entwicklung GmbH 2005 5 * 6 * Author: Arnd Bergmann <arndb@de.ibm.com> 7 * 8 * This program is free software; you can redistribute it and/or modify 9 * it under the terms of the GNU General Public License as published by 10 * the Free Software Foundation; either version 2, or (at your option) 11 * any later version. 12 * 13 * This program is distributed in the hope that it will be useful, 14 * but WITHOUT ANY WARRANTY; without even the implied warranty of 15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 16 * GNU General Public License for more details. 17 * 18 * You should have received a copy of the GNU General Public License 19 * along with this program; if not, write to the Free Software 20 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. 21 */ 22 23#include <linux/list.h> 24#include <linux/slab.h> 25 26#include "spufs.h" 27 28struct spu_gang *alloc_spu_gang(void) 29{ 30 struct spu_gang *gang; 31 32 gang = kzalloc(sizeof *gang, GFP_KERNEL); 33 if (!gang) 34 goto out; 35 36 kref_init(&gang->kref); 37 mutex_init(&gang->mutex); 38 INIT_LIST_HEAD(&gang->list); 39 40out: 41 return gang; 42} 43 44static void destroy_spu_gang(struct kref *kref) 45{ 46 struct spu_gang *gang; 47 gang = container_of(kref, struct spu_gang, kref); 48 WARN_ON(gang->contexts || !list_empty(&gang->list)); 49 kfree(gang); 50} 51 52struct spu_gang *get_spu_gang(struct spu_gang *gang) 53{ 54 kref_get(&gang->kref); 55 return gang; 56} 57 58int put_spu_gang(struct spu_gang *gang) 59{ 60 return kref_put(&gang->kref, &destroy_spu_gang); 61} 62 63void spu_gang_add_ctx(struct spu_gang *gang, struct spu_context *ctx) 64{ 65 mutex_lock(&gang->mutex); 66 ctx->gang = get_spu_gang(gang); 67 list_add(&ctx->gang_list, &gang->list); 68 gang->contexts++; 69 mutex_unlock(&gang->mutex); 70} 71 72void spu_gang_remove_ctx(struct spu_gang *gang, struct spu_context *ctx) 73{ 74 mutex_lock(&gang->mutex); 75 WARN_ON(ctx->gang != gang); 76 list_del_init(&ctx->gang_list); 77 gang->contexts--; 78 mutex_unlock(&gang->mutex); 79 80 put_spu_gang(gang); 81} 82