158 void gru_flush_tlb_range(struct gru_mm_struct *gms, unsigned long start,  in gru_flush_tlb_range()  argument
174 	gru_dbg(grudev, "gms %p, start 0x%lx, len 0x%lx, asidmap 0x%lx\n", gms,  in gru_flush_tlb_range()
175 		start, len, gms->ms_asidmap[0]);  in gru_flush_tlb_range()
177 	spin_lock(&gms->ms_asid_lock);  in gru_flush_tlb_range()
178 	for_each_gru_in_bitmap(gid, gms->ms_asidmap) {  in gru_flush_tlb_range()
181 		asids = gms->ms_asids + gid;  in gru_flush_tlb_range()
196 			__clear_bit(gru->gs_gid, gms->ms_asidmap);  in gru_flush_tlb_range()
200 				gms->ms_asidmap[0]);  in gru_flush_tlb_range()
203 	spin_unlock(&gms->ms_asid_lock);  in gru_flush_tlb_range()
226 	struct gru_mm_struct *gms = container_of(mn, struct gru_mm_struct,  in gru_invalidate_range_start()  local
230 	atomic_inc(&gms->ms_range_active);  in gru_invalidate_range_start()
231 	gru_dbg(grudev, "gms %p, start 0x%lx, end 0x%lx, act %d\n", gms,  in gru_invalidate_range_start()
232 		start, end, atomic_read(&gms->ms_range_active));  in gru_invalidate_range_start()
233 	gru_flush_tlb_range(gms, start, end - start);  in gru_invalidate_range_start()
240 	struct gru_mm_struct *gms = container_of(mn, struct gru_mm_struct,  in gru_invalidate_range_end()  local
244 	(void)atomic_dec_and_test(&gms->ms_range_active);  in gru_invalidate_range_end()
246 	wake_up_all(&gms->ms_wait_queue);  in gru_invalidate_range_end()
247 	gru_dbg(grudev, "gms %p, start 0x%lx, end 0x%lx\n", gms, start, end);  in gru_invalidate_range_end()
253 	struct gru_mm_struct *gms = container_of(mn, struct gru_mm_struct,  in gru_invalidate_page()  local
257 	gru_flush_tlb_range(gms, address, PAGE_SIZE);  in gru_invalidate_page()
258 	gru_dbg(grudev, "gms %p, address 0x%lx\n", gms, address);  in gru_invalidate_page()
263 	struct gru_mm_struct *gms = container_of(mn, struct gru_mm_struct,  in gru_release()  local
266 	gms->ms_released = 1;  in gru_release()
267 	gru_dbg(grudev, "gms %p\n", gms);  in gru_release()
299 	struct gru_mm_struct *gms;  in gru_register_mmu_notifier()  local
305 		gms = container_of(mn, struct gru_mm_struct, ms_notifier);  in gru_register_mmu_notifier()
306 		atomic_inc(&gms->ms_refcnt);  in gru_register_mmu_notifier()
308 		gms = kzalloc(sizeof(*gms), GFP_KERNEL);  in gru_register_mmu_notifier()
309 		if (!gms)  in gru_register_mmu_notifier()
312 		spin_lock_init(&gms->ms_asid_lock);  in gru_register_mmu_notifier()
313 		gms->ms_notifier.ops = &gru_mmuops;  in gru_register_mmu_notifier()
314 		atomic_set(&gms->ms_refcnt, 1);  in gru_register_mmu_notifier()
315 		init_waitqueue_head(&gms->ms_wait_queue);  in gru_register_mmu_notifier()
316 		err = __mmu_notifier_register(&gms->ms_notifier, current->mm);  in gru_register_mmu_notifier()
320 	if (gms)  in gru_register_mmu_notifier()
321 		gru_dbg(grudev, "gms %p, refcnt %d\n", gms,  in gru_register_mmu_notifier()
322 			atomic_read(&gms->ms_refcnt));  in gru_register_mmu_notifier()
323 	return gms;  in gru_register_mmu_notifier()
325 	kfree(gms);  in gru_register_mmu_notifier()
329 void gru_drop_mmu_notifier(struct gru_mm_struct *gms)  in gru_drop_mmu_notifier()  argument
331 	gru_dbg(grudev, "gms %p, refcnt %d, released %d\n", gms,  in gru_drop_mmu_notifier()
332 		atomic_read(&gms->ms_refcnt), gms->ms_released);  in gru_drop_mmu_notifier()
333 	if (atomic_dec_return(&gms->ms_refcnt) == 0) {  in gru_drop_mmu_notifier()
334 		if (!gms->ms_released)  in gru_drop_mmu_notifier()
335 			mmu_notifier_unregister(&gms->ms_notifier, current->mm);  in gru_drop_mmu_notifier()
336 		kfree(gms);  in gru_drop_mmu_notifier()