}
 
-       for_each_possible_cpu(cpu) {
+       for_each_cpu_mask(cpu, mask) {
                data = iter->tr->data[cpu];
-
-               if (!cpu_isset(cpu, mask))
-                       continue;
                spin_unlock(&data->lock);
                atomic_dec(&data->disabled);
        }
        for_each_possible_cpu(i) {
                data = global_trace.data[i];
                spin_lock_init(&data->lock);
+               lockdep_set_class(&data->lock, &data->lock_key);
                page = list_entry(pages.next, struct page, lru);
                list_del_init(&page->lru);
                list_add_tail(&page->lru, &data->trace_pages);
 #ifdef CONFIG_TRACER_MAX_TRACE
                data = max_tr.data[i];
                spin_lock_init(&data->lock);
+               lockdep_set_class(&data->lock, &data->lock_key);
                page = list_entry(pages.next, struct page, lru);
                list_del_init(&page->lru);
                list_add_tail(&page->lru, &data->trace_pages);
        }
        return ret;
 }
-
 fs_initcall(tracer_alloc_buffers);