IB/mthca: Disable tuning PCI read burst size
[safe/jmp/linux-2.6] / drivers / infiniband / hw / mthca / mthca_main.c
1 /*
2  * Copyright (c) 2004, 2005 Topspin Communications.  All rights reserved.
3  * Copyright (c) 2005 Sun Microsystems, Inc. All rights reserved.
4  * Copyright (c) 2005 Mellanox Technologies. All rights reserved.
5  *
6  * This software is available to you under a choice of one of two
7  * licenses.  You may choose to be licensed under the terms of the GNU
8  * General Public License (GPL) Version 2, available from the file
9  * COPYING in the main directory of this source tree, or the
10  * OpenIB.org BSD license below:
11  *
12  *     Redistribution and use in source and binary forms, with or
13  *     without modification, are permitted provided that the following
14  *     conditions are met:
15  *
16  *      - Redistributions of source code must retain the above
17  *        copyright notice, this list of conditions and the following
18  *        disclaimer.
19  *
20  *      - Redistributions in binary form must reproduce the above
21  *        copyright notice, this list of conditions and the following
22  *        disclaimer in the documentation and/or other materials
23  *        provided with the distribution.
24  *
25  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
26  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
27  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
28  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
29  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
30  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
31  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
32  * SOFTWARE.
33  *
34  * $Id: mthca_main.c 1396 2004-12-28 04:10:27Z roland $
35  */
36
37 #include <linux/config.h>
38 #include <linux/module.h>
39 #include <linux/init.h>
40 #include <linux/errno.h>
41 #include <linux/pci.h>
42 #include <linux/interrupt.h>
43
44 #include "mthca_dev.h"
45 #include "mthca_config_reg.h"
46 #include "mthca_cmd.h"
47 #include "mthca_profile.h"
48 #include "mthca_memfree.h"
49
50 MODULE_AUTHOR("Roland Dreier");
51 MODULE_DESCRIPTION("Mellanox InfiniBand HCA low-level driver");
52 MODULE_LICENSE("Dual BSD/GPL");
53 MODULE_VERSION(DRV_VERSION);
54
55 #ifdef CONFIG_INFINIBAND_MTHCA_DEBUG
56
57 int mthca_debug_level = 0;
58 module_param_named(debug_level, mthca_debug_level, int, 0644);
59 MODULE_PARM_DESC(debug_level, "Enable debug tracing if > 0");
60
61 #endif /* CONFIG_INFINIBAND_MTHCA_DEBUG */
62
63 #ifdef CONFIG_PCI_MSI
64
65 static int msi_x = 0;
66 module_param(msi_x, int, 0444);
67 MODULE_PARM_DESC(msi_x, "attempt to use MSI-X if nonzero");
68
69 static int msi = 0;
70 module_param(msi, int, 0444);
71 MODULE_PARM_DESC(msi, "attempt to use MSI if nonzero");
72
73 #else /* CONFIG_PCI_MSI */
74
75 #define msi_x (0)
76 #define msi   (0)
77
78 #endif /* CONFIG_PCI_MSI */
79
80 static int tune_pci = 0;
81 module_param(tune_pci, int, 0444);
82 MODULE_PARM_DESC(tune_pci, "increase PCI burst from the default set by BIOS if nonzero");
83
84 static const char mthca_version[] __devinitdata =
85         DRV_NAME ": Mellanox InfiniBand HCA driver v"
86         DRV_VERSION " (" DRV_RELDATE ")\n";
87
88 static struct mthca_profile default_profile = {
89         .num_qp            = 1 << 16,
90         .rdb_per_qp        = 4,
91         .num_cq            = 1 << 16,
92         .num_mcg           = 1 << 13,
93         .num_mpt           = 1 << 17,
94         .num_mtt           = 1 << 20,
95         .num_udav          = 1 << 15,   /* Tavor only */
96         .fmr_reserved_mtts = 1 << 18,   /* Tavor only */
97         .uarc_size         = 1 << 18,   /* Arbel only */
98 };
99
100 static int __devinit mthca_tune_pci(struct mthca_dev *mdev)
101 {
102         int cap;
103         u16 val;
104
105         if (!tune_pci)
106                 return 0;
107
108         /* First try to max out Read Byte Count */
109         cap = pci_find_capability(mdev->pdev, PCI_CAP_ID_PCIX);
110         if (cap) {
111                 if (pci_read_config_word(mdev->pdev, cap + PCI_X_CMD, &val)) {
112                         mthca_err(mdev, "Couldn't read PCI-X command register, "
113                                   "aborting.\n");
114                         return -ENODEV;
115                 }
116                 val = (val & ~PCI_X_CMD_MAX_READ) | (3 << 2);
117                 if (pci_write_config_word(mdev->pdev, cap + PCI_X_CMD, val)) {
118                         mthca_err(mdev, "Couldn't write PCI-X command register, "
119                                   "aborting.\n");
120                         return -ENODEV;
121                 }
122         } else if (!(mdev->mthca_flags & MTHCA_FLAG_PCIE))
123                 mthca_info(mdev, "No PCI-X capability, not setting RBC.\n");
124
125         cap = pci_find_capability(mdev->pdev, PCI_CAP_ID_EXP);
126         if (cap) {
127                 if (pci_read_config_word(mdev->pdev, cap + PCI_EXP_DEVCTL, &val)) {
128                         mthca_err(mdev, "Couldn't read PCI Express device control "
129                                   "register, aborting.\n");
130                         return -ENODEV;
131                 }
132                 val = (val & ~PCI_EXP_DEVCTL_READRQ) | (5 << 12);
133                 if (pci_write_config_word(mdev->pdev, cap + PCI_EXP_DEVCTL, val)) {
134                         mthca_err(mdev, "Couldn't write PCI Express device control "
135                                   "register, aborting.\n");
136                         return -ENODEV;
137                 }
138         } else if (mdev->mthca_flags & MTHCA_FLAG_PCIE)
139                 mthca_info(mdev, "No PCI Express capability, "
140                            "not setting Max Read Request Size.\n");
141
142         return 0;
143 }
144
145 static int __devinit mthca_dev_lim(struct mthca_dev *mdev, struct mthca_dev_lim *dev_lim)
146 {
147         int err;
148         u8 status;
149
150         err = mthca_QUERY_DEV_LIM(mdev, dev_lim, &status);
151         if (err) {
152                 mthca_err(mdev, "QUERY_DEV_LIM command failed, aborting.\n");
153                 return err;
154         }
155         if (status) {
156                 mthca_err(mdev, "QUERY_DEV_LIM returned status 0x%02x, "
157                           "aborting.\n", status);
158                 return -EINVAL;
159         }
160         if (dev_lim->min_page_sz > PAGE_SIZE) {
161                 mthca_err(mdev, "HCA minimum page size of %d bigger than "
162                           "kernel PAGE_SIZE of %ld, aborting.\n",
163                           dev_lim->min_page_sz, PAGE_SIZE);
164                 return -ENODEV;
165         }
166         if (dev_lim->num_ports > MTHCA_MAX_PORTS) {
167                 mthca_err(mdev, "HCA has %d ports, but we only support %d, "
168                           "aborting.\n",
169                           dev_lim->num_ports, MTHCA_MAX_PORTS);
170                 return -ENODEV;
171         }
172
173         if (dev_lim->uar_size > pci_resource_len(mdev->pdev, 2)) {
174                 mthca_err(mdev, "HCA reported UAR size of 0x%x bigger than "
175                           "PCI resource 2 size of 0x%lx, aborting.\n",
176                           dev_lim->uar_size, pci_resource_len(mdev->pdev, 2));
177                 return -ENODEV;
178         }
179
180         mdev->limits.num_ports          = dev_lim->num_ports;
181         mdev->limits.vl_cap             = dev_lim->max_vl;
182         mdev->limits.mtu_cap            = dev_lim->max_mtu;
183         mdev->limits.gid_table_len      = dev_lim->max_gids;
184         mdev->limits.pkey_table_len     = dev_lim->max_pkeys;
185         mdev->limits.local_ca_ack_delay = dev_lim->local_ca_ack_delay;
186         mdev->limits.max_sg             = dev_lim->max_sg;
187         mdev->limits.max_wqes           = dev_lim->max_qp_sz;
188         mdev->limits.max_qp_init_rdma   = dev_lim->max_requester_per_qp;
189         mdev->limits.reserved_qps       = dev_lim->reserved_qps;
190         mdev->limits.max_srq_wqes       = dev_lim->max_srq_sz;
191         mdev->limits.reserved_srqs      = dev_lim->reserved_srqs;
192         mdev->limits.reserved_eecs      = dev_lim->reserved_eecs;
193         mdev->limits.max_desc_sz        = dev_lim->max_desc_sz;
194         /*
195          * Subtract 1 from the limit because we need to allocate a
196          * spare CQE so the HCA HW can tell the difference between an
197          * empty CQ and a full CQ.
198          */
199         mdev->limits.max_cqes           = dev_lim->max_cq_sz - 1;
200         mdev->limits.reserved_cqs       = dev_lim->reserved_cqs;
201         mdev->limits.reserved_eqs       = dev_lim->reserved_eqs;
202         mdev->limits.reserved_mtts      = dev_lim->reserved_mtts;
203         mdev->limits.reserved_mrws      = dev_lim->reserved_mrws;
204         mdev->limits.reserved_uars      = dev_lim->reserved_uars;
205         mdev->limits.reserved_pds       = dev_lim->reserved_pds;
206         mdev->limits.port_width_cap     = dev_lim->max_port_width;
207         mdev->limits.page_size_cap      = ~(u32) (dev_lim->min_page_sz - 1);
208         mdev->limits.flags              = dev_lim->flags;
209         /*
210          * For old FW that doesn't return static rate support, use a
211          * value of 0x3 (only static rate values of 0 or 1 are handled),
212          * except on Sinai, where even old FW can handle static rate
213          * values of 2 and 3.
214          */
215         if (dev_lim->stat_rate_support)
216                 mdev->limits.stat_rate_support = dev_lim->stat_rate_support;
217         else if (mdev->mthca_flags & MTHCA_FLAG_SINAI_OPT)
218                 mdev->limits.stat_rate_support = 0xf;
219         else
220                 mdev->limits.stat_rate_support = 0x3;
221
222         /* IB_DEVICE_RESIZE_MAX_WR not supported by driver.
223            May be doable since hardware supports it for SRQ.
224
225            IB_DEVICE_N_NOTIFY_CQ is supported by hardware but not by driver.
226
227            IB_DEVICE_SRQ_RESIZE is supported by hardware but SRQ is not
228            supported by driver. */
229         mdev->device_cap_flags = IB_DEVICE_CHANGE_PHY_PORT |
230                 IB_DEVICE_PORT_ACTIVE_EVENT |
231                 IB_DEVICE_SYS_IMAGE_GUID |
232                 IB_DEVICE_RC_RNR_NAK_GEN;
233
234         if (dev_lim->flags & DEV_LIM_FLAG_BAD_PKEY_CNTR)
235                 mdev->device_cap_flags |= IB_DEVICE_BAD_PKEY_CNTR;
236
237         if (dev_lim->flags & DEV_LIM_FLAG_BAD_QKEY_CNTR)
238                 mdev->device_cap_flags |= IB_DEVICE_BAD_QKEY_CNTR;
239
240         if (dev_lim->flags & DEV_LIM_FLAG_RAW_MULTI)
241                 mdev->device_cap_flags |= IB_DEVICE_RAW_MULTI;
242
243         if (dev_lim->flags & DEV_LIM_FLAG_AUTO_PATH_MIG)
244                 mdev->device_cap_flags |= IB_DEVICE_AUTO_PATH_MIG;
245
246         if (dev_lim->flags & DEV_LIM_FLAG_UD_AV_PORT_ENFORCE)
247                 mdev->device_cap_flags |= IB_DEVICE_UD_AV_PORT_ENFORCE;
248
249         if (dev_lim->flags & DEV_LIM_FLAG_SRQ)
250                 mdev->mthca_flags |= MTHCA_FLAG_SRQ;
251
252         return 0;
253 }
254
255 static int __devinit mthca_init_tavor(struct mthca_dev *mdev)
256 {
257         u8 status;
258         int err;
259         struct mthca_dev_lim        dev_lim;
260         struct mthca_profile        profile;
261         struct mthca_init_hca_param init_hca;
262
263         err = mthca_SYS_EN(mdev, &status);
264         if (err) {
265                 mthca_err(mdev, "SYS_EN command failed, aborting.\n");
266                 return err;
267         }
268         if (status) {
269                 mthca_err(mdev, "SYS_EN returned status 0x%02x, "
270                           "aborting.\n", status);
271                 return -EINVAL;
272         }
273
274         err = mthca_QUERY_FW(mdev, &status);
275         if (err) {
276                 mthca_err(mdev, "QUERY_FW command failed, aborting.\n");
277                 goto err_disable;
278         }
279         if (status) {
280                 mthca_err(mdev, "QUERY_FW returned status 0x%02x, "
281                           "aborting.\n", status);
282                 err = -EINVAL;
283                 goto err_disable;
284         }
285         err = mthca_QUERY_DDR(mdev, &status);
286         if (err) {
287                 mthca_err(mdev, "QUERY_DDR command failed, aborting.\n");
288                 goto err_disable;
289         }
290         if (status) {
291                 mthca_err(mdev, "QUERY_DDR returned status 0x%02x, "
292                           "aborting.\n", status);
293                 err = -EINVAL;
294                 goto err_disable;
295         }
296
297         err = mthca_dev_lim(mdev, &dev_lim);
298         if (err) {
299                 mthca_err(mdev, "QUERY_DEV_LIM command failed, aborting.\n");
300                 goto err_disable;
301         }
302
303         profile = default_profile;
304         profile.num_uar   = dev_lim.uar_size / PAGE_SIZE;
305         profile.uarc_size = 0;
306         if (mdev->mthca_flags & MTHCA_FLAG_SRQ)
307                 profile.num_srq = dev_lim.max_srqs;
308
309         err = mthca_make_profile(mdev, &profile, &dev_lim, &init_hca);
310         if (err < 0)
311                 goto err_disable;
312
313         err = mthca_INIT_HCA(mdev, &init_hca, &status);
314         if (err) {
315                 mthca_err(mdev, "INIT_HCA command failed, aborting.\n");
316                 goto err_disable;
317         }
318         if (status) {
319                 mthca_err(mdev, "INIT_HCA returned status 0x%02x, "
320                           "aborting.\n", status);
321                 err = -EINVAL;
322                 goto err_disable;
323         }
324
325         return 0;
326
327 err_disable:
328         mthca_SYS_DIS(mdev, &status);
329
330         return err;
331 }
332
333 static int __devinit mthca_load_fw(struct mthca_dev *mdev)
334 {
335         u8 status;
336         int err;
337
338         /* FIXME: use HCA-attached memory for FW if present */
339
340         mdev->fw.arbel.fw_icm =
341                 mthca_alloc_icm(mdev, mdev->fw.arbel.fw_pages,
342                                 GFP_HIGHUSER | __GFP_NOWARN);
343         if (!mdev->fw.arbel.fw_icm) {
344                 mthca_err(mdev, "Couldn't allocate FW area, aborting.\n");
345                 return -ENOMEM;
346         }
347
348         err = mthca_MAP_FA(mdev, mdev->fw.arbel.fw_icm, &status);
349         if (err) {
350                 mthca_err(mdev, "MAP_FA command failed, aborting.\n");
351                 goto err_free;
352         }
353         if (status) {
354                 mthca_err(mdev, "MAP_FA returned status 0x%02x, aborting.\n", status);
355                 err = -EINVAL;
356                 goto err_free;
357         }
358         err = mthca_RUN_FW(mdev, &status);
359         if (err) {
360                 mthca_err(mdev, "RUN_FW command failed, aborting.\n");
361                 goto err_unmap_fa;
362         }
363         if (status) {
364                 mthca_err(mdev, "RUN_FW returned status 0x%02x, aborting.\n", status);
365                 err = -EINVAL;
366                 goto err_unmap_fa;
367         }
368
369         return 0;
370
371 err_unmap_fa:
372         mthca_UNMAP_FA(mdev, &status);
373
374 err_free:
375         mthca_free_icm(mdev, mdev->fw.arbel.fw_icm);
376         return err;
377 }
378
379 static int __devinit mthca_init_icm(struct mthca_dev *mdev,
380                                     struct mthca_dev_lim *dev_lim,
381                                     struct mthca_init_hca_param *init_hca,
382                                     u64 icm_size)
383 {
384         u64 aux_pages;
385         u8 status;
386         int err;
387
388         err = mthca_SET_ICM_SIZE(mdev, icm_size, &aux_pages, &status);
389         if (err) {
390                 mthca_err(mdev, "SET_ICM_SIZE command failed, aborting.\n");
391                 return err;
392         }
393         if (status) {
394                 mthca_err(mdev, "SET_ICM_SIZE returned status 0x%02x, "
395                           "aborting.\n", status);
396                 return -EINVAL;
397         }
398
399         mthca_dbg(mdev, "%lld KB of HCA context requires %lld KB aux memory.\n",
400                   (unsigned long long) icm_size >> 10,
401                   (unsigned long long) aux_pages << 2);
402
403         mdev->fw.arbel.aux_icm = mthca_alloc_icm(mdev, aux_pages,
404                                                  GFP_HIGHUSER | __GFP_NOWARN);
405         if (!mdev->fw.arbel.aux_icm) {
406                 mthca_err(mdev, "Couldn't allocate aux memory, aborting.\n");
407                 return -ENOMEM;
408         }
409
410         err = mthca_MAP_ICM_AUX(mdev, mdev->fw.arbel.aux_icm, &status);
411         if (err) {
412                 mthca_err(mdev, "MAP_ICM_AUX command failed, aborting.\n");
413                 goto err_free_aux;
414         }
415         if (status) {
416                 mthca_err(mdev, "MAP_ICM_AUX returned status 0x%02x, aborting.\n", status);
417                 err = -EINVAL;
418                 goto err_free_aux;
419         }
420
421         err = mthca_map_eq_icm(mdev, init_hca->eqc_base);
422         if (err) {
423                 mthca_err(mdev, "Failed to map EQ context memory, aborting.\n");
424                 goto err_unmap_aux;
425         }
426
427         mdev->mr_table.mtt_table = mthca_alloc_icm_table(mdev, init_hca->mtt_base,
428                                                          MTHCA_MTT_SEG_SIZE,
429                                                          mdev->limits.num_mtt_segs,
430                                                          mdev->limits.reserved_mtts, 1);
431         if (!mdev->mr_table.mtt_table) {
432                 mthca_err(mdev, "Failed to map MTT context memory, aborting.\n");
433                 err = -ENOMEM;
434                 goto err_unmap_eq;
435         }
436
437         mdev->mr_table.mpt_table = mthca_alloc_icm_table(mdev, init_hca->mpt_base,
438                                                          dev_lim->mpt_entry_sz,
439                                                          mdev->limits.num_mpts,
440                                                          mdev->limits.reserved_mrws, 1);
441         if (!mdev->mr_table.mpt_table) {
442                 mthca_err(mdev, "Failed to map MPT context memory, aborting.\n");
443                 err = -ENOMEM;
444                 goto err_unmap_mtt;
445         }
446
447         mdev->qp_table.qp_table = mthca_alloc_icm_table(mdev, init_hca->qpc_base,
448                                                         dev_lim->qpc_entry_sz,
449                                                         mdev->limits.num_qps,
450                                                         mdev->limits.reserved_qps, 0);
451         if (!mdev->qp_table.qp_table) {
452                 mthca_err(mdev, "Failed to map QP context memory, aborting.\n");
453                 err = -ENOMEM;
454                 goto err_unmap_mpt;
455         }
456
457         mdev->qp_table.eqp_table = mthca_alloc_icm_table(mdev, init_hca->eqpc_base,
458                                                          dev_lim->eqpc_entry_sz,
459                                                          mdev->limits.num_qps,
460                                                          mdev->limits.reserved_qps, 0);
461         if (!mdev->qp_table.eqp_table) {
462                 mthca_err(mdev, "Failed to map EQP context memory, aborting.\n");
463                 err = -ENOMEM;
464                 goto err_unmap_qp;
465         }
466
467         mdev->qp_table.rdb_table = mthca_alloc_icm_table(mdev, init_hca->rdb_base,
468                                                          MTHCA_RDB_ENTRY_SIZE,
469                                                          mdev->limits.num_qps <<
470                                                          mdev->qp_table.rdb_shift,
471                                                          0, 0);
472         if (!mdev->qp_table.rdb_table) {
473                 mthca_err(mdev, "Failed to map RDB context memory, aborting\n");
474                 err = -ENOMEM;
475                 goto err_unmap_eqp;
476         }
477
478        mdev->cq_table.table = mthca_alloc_icm_table(mdev, init_hca->cqc_base,
479                                                     dev_lim->cqc_entry_sz,
480                                                     mdev->limits.num_cqs,
481                                                     mdev->limits.reserved_cqs, 0);
482         if (!mdev->cq_table.table) {
483                 mthca_err(mdev, "Failed to map CQ context memory, aborting.\n");
484                 err = -ENOMEM;
485                 goto err_unmap_rdb;
486         }
487
488         if (mdev->mthca_flags & MTHCA_FLAG_SRQ) {
489                 mdev->srq_table.table =
490                         mthca_alloc_icm_table(mdev, init_hca->srqc_base,
491                                               dev_lim->srq_entry_sz,
492                                               mdev->limits.num_srqs,
493                                               mdev->limits.reserved_srqs, 0);
494                 if (!mdev->srq_table.table) {
495                         mthca_err(mdev, "Failed to map SRQ context memory, "
496                                   "aborting.\n");
497                         err = -ENOMEM;
498                         goto err_unmap_cq;
499                 }
500         }
501
502         /*
503          * It's not strictly required, but for simplicity just map the
504          * whole multicast group table now.  The table isn't very big
505          * and it's a lot easier than trying to track ref counts.
506          */
507         mdev->mcg_table.table = mthca_alloc_icm_table(mdev, init_hca->mc_base,
508                                                       MTHCA_MGM_ENTRY_SIZE,
509                                                       mdev->limits.num_mgms +
510                                                       mdev->limits.num_amgms,
511                                                       mdev->limits.num_mgms +
512                                                       mdev->limits.num_amgms,
513                                                       0);
514         if (!mdev->mcg_table.table) {
515                 mthca_err(mdev, "Failed to map MCG context memory, aborting.\n");
516                 err = -ENOMEM;
517                 goto err_unmap_srq;
518         }
519
520         return 0;
521
522 err_unmap_srq:
523         if (mdev->mthca_flags & MTHCA_FLAG_SRQ)
524                 mthca_free_icm_table(mdev, mdev->srq_table.table);
525
526 err_unmap_cq:
527         mthca_free_icm_table(mdev, mdev->cq_table.table);
528
529 err_unmap_rdb:
530         mthca_free_icm_table(mdev, mdev->qp_table.rdb_table);
531
532 err_unmap_eqp:
533         mthca_free_icm_table(mdev, mdev->qp_table.eqp_table);
534
535 err_unmap_qp:
536         mthca_free_icm_table(mdev, mdev->qp_table.qp_table);
537
538 err_unmap_mpt:
539         mthca_free_icm_table(mdev, mdev->mr_table.mpt_table);
540
541 err_unmap_mtt:
542         mthca_free_icm_table(mdev, mdev->mr_table.mtt_table);
543
544 err_unmap_eq:
545         mthca_unmap_eq_icm(mdev);
546
547 err_unmap_aux:
548         mthca_UNMAP_ICM_AUX(mdev, &status);
549
550 err_free_aux:
551         mthca_free_icm(mdev, mdev->fw.arbel.aux_icm);
552
553         return err;
554 }
555
556 static void mthca_free_icms(struct mthca_dev *mdev)
557 {
558         u8 status;
559
560         mthca_free_icm_table(mdev, mdev->mcg_table.table);
561         if (mdev->mthca_flags & MTHCA_FLAG_SRQ)
562                 mthca_free_icm_table(mdev, mdev->srq_table.table);
563         mthca_free_icm_table(mdev, mdev->cq_table.table);
564         mthca_free_icm_table(mdev, mdev->qp_table.rdb_table);
565         mthca_free_icm_table(mdev, mdev->qp_table.eqp_table);
566         mthca_free_icm_table(mdev, mdev->qp_table.qp_table);
567         mthca_free_icm_table(mdev, mdev->mr_table.mpt_table);
568         mthca_free_icm_table(mdev, mdev->mr_table.mtt_table);
569         mthca_unmap_eq_icm(mdev);
570
571         mthca_UNMAP_ICM_AUX(mdev, &status);
572         mthca_free_icm(mdev, mdev->fw.arbel.aux_icm);
573 }
574
575 static int __devinit mthca_init_arbel(struct mthca_dev *mdev)
576 {
577         struct mthca_dev_lim        dev_lim;
578         struct mthca_profile        profile;
579         struct mthca_init_hca_param init_hca;
580         u64 icm_size;
581         u8 status;
582         int err;
583
584         err = mthca_QUERY_FW(mdev, &status);
585         if (err) {
586                 mthca_err(mdev, "QUERY_FW command failed, aborting.\n");
587                 return err;
588         }
589         if (status) {
590                 mthca_err(mdev, "QUERY_FW returned status 0x%02x, "
591                           "aborting.\n", status);
592                 return -EINVAL;
593         }
594
595         err = mthca_ENABLE_LAM(mdev, &status);
596         if (err) {
597                 mthca_err(mdev, "ENABLE_LAM command failed, aborting.\n");
598                 return err;
599         }
600         if (status == MTHCA_CMD_STAT_LAM_NOT_PRE) {
601                 mthca_dbg(mdev, "No HCA-attached memory (running in MemFree mode)\n");
602                 mdev->mthca_flags |= MTHCA_FLAG_NO_LAM;
603         } else if (status) {
604                 mthca_err(mdev, "ENABLE_LAM returned status 0x%02x, "
605                           "aborting.\n", status);
606                 return -EINVAL;
607         }
608
609         err = mthca_load_fw(mdev);
610         if (err) {
611                 mthca_err(mdev, "Failed to start FW, aborting.\n");
612                 goto err_disable;
613         }
614
615         err = mthca_dev_lim(mdev, &dev_lim);
616         if (err) {
617                 mthca_err(mdev, "QUERY_DEV_LIM command failed, aborting.\n");
618                 goto err_stop_fw;
619         }
620
621         profile = default_profile;
622         profile.num_uar  = dev_lim.uar_size / PAGE_SIZE;
623         profile.num_udav = 0;
624         if (mdev->mthca_flags & MTHCA_FLAG_SRQ)
625                 profile.num_srq = dev_lim.max_srqs;
626
627         icm_size = mthca_make_profile(mdev, &profile, &dev_lim, &init_hca);
628         if ((int) icm_size < 0) {
629                 err = icm_size;
630                 goto err_stop_fw;
631         }
632
633         err = mthca_init_icm(mdev, &dev_lim, &init_hca, icm_size);
634         if (err)
635                 goto err_stop_fw;
636
637         err = mthca_INIT_HCA(mdev, &init_hca, &status);
638         if (err) {
639                 mthca_err(mdev, "INIT_HCA command failed, aborting.\n");
640                 goto err_free_icm;
641         }
642         if (status) {
643                 mthca_err(mdev, "INIT_HCA returned status 0x%02x, "
644                           "aborting.\n", status);
645                 err = -EINVAL;
646                 goto err_free_icm;
647         }
648
649         return 0;
650
651 err_free_icm:
652         mthca_free_icms(mdev);
653
654 err_stop_fw:
655         mthca_UNMAP_FA(mdev, &status);
656         mthca_free_icm(mdev, mdev->fw.arbel.fw_icm);
657
658 err_disable:
659         if (!(mdev->mthca_flags & MTHCA_FLAG_NO_LAM))
660                 mthca_DISABLE_LAM(mdev, &status);
661
662         return err;
663 }
664
665 static void mthca_close_hca(struct mthca_dev *mdev)
666 {
667         u8 status;
668
669         mthca_CLOSE_HCA(mdev, 0, &status);
670
671         if (mthca_is_memfree(mdev)) {
672                 mthca_free_icms(mdev);
673
674                 mthca_UNMAP_FA(mdev, &status);
675                 mthca_free_icm(mdev, mdev->fw.arbel.fw_icm);
676
677                 if (!(mdev->mthca_flags & MTHCA_FLAG_NO_LAM))
678                         mthca_DISABLE_LAM(mdev, &status);
679         } else
680                 mthca_SYS_DIS(mdev, &status);
681 }
682
683 static int __devinit mthca_init_hca(struct mthca_dev *mdev)
684 {
685         u8 status;
686         int err;
687         struct mthca_adapter adapter;
688
689         if (mthca_is_memfree(mdev))
690                 err = mthca_init_arbel(mdev);
691         else
692                 err = mthca_init_tavor(mdev);
693
694         if (err)
695                 return err;
696
697         err = mthca_QUERY_ADAPTER(mdev, &adapter, &status);
698         if (err) {
699                 mthca_err(mdev, "QUERY_ADAPTER command failed, aborting.\n");
700                 goto err_close;
701         }
702         if (status) {
703                 mthca_err(mdev, "QUERY_ADAPTER returned status 0x%02x, "
704                           "aborting.\n", status);
705                 err = -EINVAL;
706                 goto err_close;
707         }
708
709         mdev->eq_table.inta_pin = adapter.inta_pin;
710         mdev->rev_id            = adapter.revision_id;
711         memcpy(mdev->board_id, adapter.board_id, sizeof mdev->board_id);
712
713         return 0;
714
715 err_close:
716         mthca_close_hca(mdev);
717         return err;
718 }
719
720 static int __devinit mthca_setup_hca(struct mthca_dev *dev)
721 {
722         int err;
723         u8 status;
724
725         MTHCA_INIT_DOORBELL_LOCK(&dev->doorbell_lock);
726
727         err = mthca_init_uar_table(dev);
728         if (err) {
729                 mthca_err(dev, "Failed to initialize "
730                           "user access region table, aborting.\n");
731                 return err;
732         }
733
734         err = mthca_uar_alloc(dev, &dev->driver_uar);
735         if (err) {
736                 mthca_err(dev, "Failed to allocate driver access region, "
737                           "aborting.\n");
738                 goto err_uar_table_free;
739         }
740
741         dev->kar = ioremap(dev->driver_uar.pfn << PAGE_SHIFT, PAGE_SIZE);
742         if (!dev->kar) {
743                 mthca_err(dev, "Couldn't map kernel access region, "
744                           "aborting.\n");
745                 err = -ENOMEM;
746                 goto err_uar_free;
747         }
748
749         err = mthca_init_pd_table(dev);
750         if (err) {
751                 mthca_err(dev, "Failed to initialize "
752                           "protection domain table, aborting.\n");
753                 goto err_kar_unmap;
754         }
755
756         err = mthca_init_mr_table(dev);
757         if (err) {
758                 mthca_err(dev, "Failed to initialize "
759                           "memory region table, aborting.\n");
760                 goto err_pd_table_free;
761         }
762
763         err = mthca_pd_alloc(dev, 1, &dev->driver_pd);
764         if (err) {
765                 mthca_err(dev, "Failed to create driver PD, "
766                           "aborting.\n");
767                 goto err_mr_table_free;
768         }
769
770         err = mthca_init_eq_table(dev);
771         if (err) {
772                 mthca_err(dev, "Failed to initialize "
773                           "event queue table, aborting.\n");
774                 goto err_pd_free;
775         }
776
777         err = mthca_cmd_use_events(dev);
778         if (err) {
779                 mthca_err(dev, "Failed to switch to event-driven "
780                           "firmware commands, aborting.\n");
781                 goto err_eq_table_free;
782         }
783
784         err = mthca_NOP(dev, &status);
785         if (err || status) {
786                 mthca_err(dev, "NOP command failed to generate interrupt (IRQ %d), aborting.\n",
787                           dev->mthca_flags & MTHCA_FLAG_MSI_X ?
788                           dev->eq_table.eq[MTHCA_EQ_CMD].msi_x_vector :
789                           dev->pdev->irq);
790                 if (dev->mthca_flags & (MTHCA_FLAG_MSI | MTHCA_FLAG_MSI_X))
791                         mthca_err(dev, "Try again with MSI/MSI-X disabled.\n");
792                 else
793                         mthca_err(dev, "BIOS or ACPI interrupt routing problem?\n");
794
795                 goto err_cmd_poll;
796         }
797
798         mthca_dbg(dev, "NOP command IRQ test passed\n");
799
800         err = mthca_init_cq_table(dev);
801         if (err) {
802                 mthca_err(dev, "Failed to initialize "
803                           "completion queue table, aborting.\n");
804                 goto err_cmd_poll;
805         }
806
807         err = mthca_init_srq_table(dev);
808         if (err) {
809                 mthca_err(dev, "Failed to initialize "
810                           "shared receive queue table, aborting.\n");
811                 goto err_cq_table_free;
812         }
813
814         err = mthca_init_qp_table(dev);
815         if (err) {
816                 mthca_err(dev, "Failed to initialize "
817                           "queue pair table, aborting.\n");
818                 goto err_srq_table_free;
819         }
820
821         err = mthca_init_av_table(dev);
822         if (err) {
823                 mthca_err(dev, "Failed to initialize "
824                           "address vector table, aborting.\n");
825                 goto err_qp_table_free;
826         }
827
828         err = mthca_init_mcg_table(dev);
829         if (err) {
830                 mthca_err(dev, "Failed to initialize "
831                           "multicast group table, aborting.\n");
832                 goto err_av_table_free;
833         }
834
835         return 0;
836
837 err_av_table_free:
838         mthca_cleanup_av_table(dev);
839
840 err_qp_table_free:
841         mthca_cleanup_qp_table(dev);
842
843 err_srq_table_free:
844         mthca_cleanup_srq_table(dev);
845
846 err_cq_table_free:
847         mthca_cleanup_cq_table(dev);
848
849 err_cmd_poll:
850         mthca_cmd_use_polling(dev);
851
852 err_eq_table_free:
853         mthca_cleanup_eq_table(dev);
854
855 err_pd_free:
856         mthca_pd_free(dev, &dev->driver_pd);
857
858 err_mr_table_free:
859         mthca_cleanup_mr_table(dev);
860
861 err_pd_table_free:
862         mthca_cleanup_pd_table(dev);
863
864 err_kar_unmap:
865         iounmap(dev->kar);
866
867 err_uar_free:
868         mthca_uar_free(dev, &dev->driver_uar);
869
870 err_uar_table_free:
871         mthca_cleanup_uar_table(dev);
872         return err;
873 }
874
875 static int __devinit mthca_request_regions(struct pci_dev *pdev,
876                                            int ddr_hidden)
877 {
878         int err;
879
880         /*
881          * We can't just use pci_request_regions() because the MSI-X
882          * table is right in the middle of the first BAR.  If we did
883          * pci_request_region and grab all of the first BAR, then
884          * setting up MSI-X would fail, since the PCI core wants to do
885          * request_mem_region on the MSI-X vector table.
886          *
887          * So just request what we need right now, and request any
888          * other regions we need when setting up EQs.
889          */
890         if (!request_mem_region(pci_resource_start(pdev, 0) + MTHCA_HCR_BASE,
891                                 MTHCA_HCR_SIZE, DRV_NAME))
892                 return -EBUSY;
893
894         err = pci_request_region(pdev, 2, DRV_NAME);
895         if (err)
896                 goto err_bar2_failed;
897
898         if (!ddr_hidden) {
899                 err = pci_request_region(pdev, 4, DRV_NAME);
900                 if (err)
901                         goto err_bar4_failed;
902         }
903
904         return 0;
905
906 err_bar4_failed:
907         pci_release_region(pdev, 2);
908
909 err_bar2_failed:
910         release_mem_region(pci_resource_start(pdev, 0) + MTHCA_HCR_BASE,
911                            MTHCA_HCR_SIZE);
912
913         return err;
914 }
915
916 static void mthca_release_regions(struct pci_dev *pdev,
917                                   int ddr_hidden)
918 {
919         if (!ddr_hidden)
920                 pci_release_region(pdev, 4);
921
922         pci_release_region(pdev, 2);
923
924         release_mem_region(pci_resource_start(pdev, 0) + MTHCA_HCR_BASE,
925                            MTHCA_HCR_SIZE);
926 }
927
928 static int __devinit mthca_enable_msi_x(struct mthca_dev *mdev)
929 {
930         struct msix_entry entries[3];
931         int err;
932
933         entries[0].entry = 0;
934         entries[1].entry = 1;
935         entries[2].entry = 2;
936
937         err = pci_enable_msix(mdev->pdev, entries, ARRAY_SIZE(entries));
938         if (err) {
939                 if (err > 0)
940                         mthca_info(mdev, "Only %d MSI-X vectors available, "
941                                    "not using MSI-X\n", err);
942                 return err;
943         }
944
945         mdev->eq_table.eq[MTHCA_EQ_COMP ].msi_x_vector = entries[0].vector;
946         mdev->eq_table.eq[MTHCA_EQ_ASYNC].msi_x_vector = entries[1].vector;
947         mdev->eq_table.eq[MTHCA_EQ_CMD  ].msi_x_vector = entries[2].vector;
948
949         return 0;
950 }
951
952 /* Types of supported HCA */
953 enum {
954         TAVOR,                  /* MT23108                        */
955         ARBEL_COMPAT,           /* MT25208 in Tavor compat mode   */
956         ARBEL_NATIVE,           /* MT25208 with extended features */
957         SINAI                   /* MT25204 */
958 };
959
960 #define MTHCA_FW_VER(major, minor, subminor) \
961         (((u64) (major) << 32) | ((u64) (minor) << 16) | (u64) (subminor))
962
963 static struct {
964         u64 latest_fw;
965         u32 flags;
966 } mthca_hca_table[] = {
967         [TAVOR]        = { .latest_fw = MTHCA_FW_VER(3, 4, 0),
968                            .flags     = 0 },
969         [ARBEL_COMPAT] = { .latest_fw = MTHCA_FW_VER(4, 7, 400),
970                            .flags     = MTHCA_FLAG_PCIE },
971         [ARBEL_NATIVE] = { .latest_fw = MTHCA_FW_VER(5, 1, 0),
972                            .flags     = MTHCA_FLAG_MEMFREE |
973                                         MTHCA_FLAG_PCIE },
974         [SINAI]        = { .latest_fw = MTHCA_FW_VER(1, 0, 800),
975                            .flags     = MTHCA_FLAG_MEMFREE |
976                                         MTHCA_FLAG_PCIE    |
977                                         MTHCA_FLAG_SINAI_OPT }
978 };
979
980 static int __devinit mthca_init_one(struct pci_dev *pdev,
981                                     const struct pci_device_id *id)
982 {
983         static int mthca_version_printed = 0;
984         int ddr_hidden = 0;
985         int err;
986         struct mthca_dev *mdev;
987
988         if (!mthca_version_printed) {
989                 printk(KERN_INFO "%s", mthca_version);
990                 ++mthca_version_printed;
991         }
992
993         printk(KERN_INFO PFX "Initializing %s\n",
994                pci_name(pdev));
995
996         if (id->driver_data >= ARRAY_SIZE(mthca_hca_table)) {
997                 printk(KERN_ERR PFX "%s has invalid driver data %lx\n",
998                        pci_name(pdev), id->driver_data);
999                 return -ENODEV;
1000         }
1001
1002         err = pci_enable_device(pdev);
1003         if (err) {
1004                 dev_err(&pdev->dev, "Cannot enable PCI device, "
1005                         "aborting.\n");
1006                 return err;
1007         }
1008
1009         /*
1010          * Check for BARs.  We expect 0: 1MB, 2: 8MB, 4: DDR (may not
1011          * be present)
1012          */
1013         if (!(pci_resource_flags(pdev, 0) & IORESOURCE_MEM) ||
1014             pci_resource_len(pdev, 0) != 1 << 20) {
1015                 dev_err(&pdev->dev, "Missing DCS, aborting.\n");
1016                 err = -ENODEV;
1017                 goto err_disable_pdev;
1018         }
1019         if (!(pci_resource_flags(pdev, 2) & IORESOURCE_MEM)) {
1020                 dev_err(&pdev->dev, "Missing UAR, aborting.\n");
1021                 err = -ENODEV;
1022                 goto err_disable_pdev;
1023         }
1024         if (!(pci_resource_flags(pdev, 4) & IORESOURCE_MEM))
1025                 ddr_hidden = 1;
1026
1027         err = mthca_request_regions(pdev, ddr_hidden);
1028         if (err) {
1029                 dev_err(&pdev->dev, "Cannot obtain PCI resources, "
1030                         "aborting.\n");
1031                 goto err_disable_pdev;
1032         }
1033
1034         pci_set_master(pdev);
1035
1036         err = pci_set_dma_mask(pdev, DMA_64BIT_MASK);
1037         if (err) {
1038                 dev_warn(&pdev->dev, "Warning: couldn't set 64-bit PCI DMA mask.\n");
1039                 err = pci_set_dma_mask(pdev, DMA_32BIT_MASK);
1040                 if (err) {
1041                         dev_err(&pdev->dev, "Can't set PCI DMA mask, aborting.\n");
1042                         goto err_free_res;
1043                 }
1044         }
1045         err = pci_set_consistent_dma_mask(pdev, DMA_64BIT_MASK);
1046         if (err) {
1047                 dev_warn(&pdev->dev, "Warning: couldn't set 64-bit "
1048                          "consistent PCI DMA mask.\n");
1049                 err = pci_set_consistent_dma_mask(pdev, DMA_32BIT_MASK);
1050                 if (err) {
1051                         dev_err(&pdev->dev, "Can't set consistent PCI DMA mask, "
1052                                 "aborting.\n");
1053                         goto err_free_res;
1054                 }
1055         }
1056
1057         mdev = (struct mthca_dev *) ib_alloc_device(sizeof *mdev);
1058         if (!mdev) {
1059                 dev_err(&pdev->dev, "Device struct alloc failed, "
1060                         "aborting.\n");
1061                 err = -ENOMEM;
1062                 goto err_free_res;
1063         }
1064
1065         mdev->pdev = pdev;
1066
1067         mdev->mthca_flags = mthca_hca_table[id->driver_data].flags;
1068         if (ddr_hidden)
1069                 mdev->mthca_flags |= MTHCA_FLAG_DDR_HIDDEN;
1070
1071         /*
1072          * Now reset the HCA before we touch the PCI capabilities or
1073          * attempt a firmware command, since a boot ROM may have left
1074          * the HCA in an undefined state.
1075          */
1076         err = mthca_reset(mdev);
1077         if (err) {
1078                 mthca_err(mdev, "Failed to reset HCA, aborting.\n");
1079                 goto err_free_dev;
1080         }
1081
1082         if (msi_x && !mthca_enable_msi_x(mdev))
1083                 mdev->mthca_flags |= MTHCA_FLAG_MSI_X;
1084         if (msi && !(mdev->mthca_flags & MTHCA_FLAG_MSI_X) &&
1085             !pci_enable_msi(pdev))
1086                 mdev->mthca_flags |= MTHCA_FLAG_MSI;
1087
1088         if (mthca_cmd_init(mdev)) {
1089                 mthca_err(mdev, "Failed to init command interface, aborting.\n");
1090                 goto err_free_dev;
1091         }
1092
1093         err = mthca_tune_pci(mdev);
1094         if (err)
1095                 goto err_cmd;
1096
1097         err = mthca_init_hca(mdev);
1098         if (err)
1099                 goto err_cmd;
1100
1101         if (mdev->fw_ver < mthca_hca_table[id->driver_data].latest_fw) {
1102                 mthca_warn(mdev, "HCA FW version %d.%d.%d is old (%d.%d.%d is current).\n",
1103                            (int) (mdev->fw_ver >> 32), (int) (mdev->fw_ver >> 16) & 0xffff,
1104                            (int) (mdev->fw_ver & 0xffff),
1105                            (int) (mthca_hca_table[id->driver_data].latest_fw >> 32),
1106                            (int) (mthca_hca_table[id->driver_data].latest_fw >> 16) & 0xffff,
1107                            (int) (mthca_hca_table[id->driver_data].latest_fw & 0xffff));
1108                 mthca_warn(mdev, "If you have problems, try updating your HCA FW.\n");
1109         }
1110
1111         err = mthca_setup_hca(mdev);
1112         if (err)
1113                 goto err_close;
1114
1115         err = mthca_register_device(mdev);
1116         if (err)
1117                 goto err_cleanup;
1118
1119         err = mthca_create_agents(mdev);
1120         if (err)
1121                 goto err_unregister;
1122
1123         pci_set_drvdata(pdev, mdev);
1124
1125         return 0;
1126
1127 err_unregister:
1128         mthca_unregister_device(mdev);
1129
1130 err_cleanup:
1131         mthca_cleanup_mcg_table(mdev);
1132         mthca_cleanup_av_table(mdev);
1133         mthca_cleanup_qp_table(mdev);
1134         mthca_cleanup_srq_table(mdev);
1135         mthca_cleanup_cq_table(mdev);
1136         mthca_cmd_use_polling(mdev);
1137         mthca_cleanup_eq_table(mdev);
1138
1139         mthca_pd_free(mdev, &mdev->driver_pd);
1140
1141         mthca_cleanup_mr_table(mdev);
1142         mthca_cleanup_pd_table(mdev);
1143         mthca_cleanup_uar_table(mdev);
1144
1145 err_close:
1146         mthca_close_hca(mdev);
1147
1148 err_cmd:
1149         mthca_cmd_cleanup(mdev);
1150
1151 err_free_dev:
1152         if (mdev->mthca_flags & MTHCA_FLAG_MSI_X)
1153                 pci_disable_msix(pdev);
1154         if (mdev->mthca_flags & MTHCA_FLAG_MSI)
1155                 pci_disable_msi(pdev);
1156
1157         ib_dealloc_device(&mdev->ib_dev);
1158
1159 err_free_res:
1160         mthca_release_regions(pdev, ddr_hidden);
1161
1162 err_disable_pdev:
1163         pci_disable_device(pdev);
1164         pci_set_drvdata(pdev, NULL);
1165         return err;
1166 }
1167
1168 static void __devexit mthca_remove_one(struct pci_dev *pdev)
1169 {
1170         struct mthca_dev *mdev = pci_get_drvdata(pdev);
1171         u8 status;
1172         int p;
1173
1174         if (mdev) {
1175                 mthca_free_agents(mdev);
1176                 mthca_unregister_device(mdev);
1177
1178                 for (p = 1; p <= mdev->limits.num_ports; ++p)
1179                         mthca_CLOSE_IB(mdev, p, &status);
1180
1181                 mthca_cleanup_mcg_table(mdev);
1182                 mthca_cleanup_av_table(mdev);
1183                 mthca_cleanup_qp_table(mdev);
1184                 mthca_cleanup_srq_table(mdev);
1185                 mthca_cleanup_cq_table(mdev);
1186                 mthca_cmd_use_polling(mdev);
1187                 mthca_cleanup_eq_table(mdev);
1188
1189                 mthca_pd_free(mdev, &mdev->driver_pd);
1190
1191                 mthca_cleanup_mr_table(mdev);
1192                 mthca_cleanup_pd_table(mdev);
1193
1194                 iounmap(mdev->kar);
1195                 mthca_uar_free(mdev, &mdev->driver_uar);
1196                 mthca_cleanup_uar_table(mdev);
1197                 mthca_close_hca(mdev);
1198                 mthca_cmd_cleanup(mdev);
1199
1200                 if (mdev->mthca_flags & MTHCA_FLAG_MSI_X)
1201                         pci_disable_msix(pdev);
1202                 if (mdev->mthca_flags & MTHCA_FLAG_MSI)
1203                         pci_disable_msi(pdev);
1204
1205                 ib_dealloc_device(&mdev->ib_dev);
1206                 mthca_release_regions(pdev, mdev->mthca_flags &
1207                                       MTHCA_FLAG_DDR_HIDDEN);
1208                 pci_disable_device(pdev);
1209                 pci_set_drvdata(pdev, NULL);
1210         }
1211 }
1212
1213 static struct pci_device_id mthca_pci_table[] = {
1214         { PCI_DEVICE(PCI_VENDOR_ID_MELLANOX, PCI_DEVICE_ID_MELLANOX_TAVOR),
1215           .driver_data = TAVOR },
1216         { PCI_DEVICE(PCI_VENDOR_ID_TOPSPIN, PCI_DEVICE_ID_MELLANOX_TAVOR),
1217           .driver_data = TAVOR },
1218         { PCI_DEVICE(PCI_VENDOR_ID_MELLANOX, PCI_DEVICE_ID_MELLANOX_ARBEL_COMPAT),
1219           .driver_data = ARBEL_COMPAT },
1220         { PCI_DEVICE(PCI_VENDOR_ID_TOPSPIN, PCI_DEVICE_ID_MELLANOX_ARBEL_COMPAT),
1221           .driver_data = ARBEL_COMPAT },
1222         { PCI_DEVICE(PCI_VENDOR_ID_MELLANOX, PCI_DEVICE_ID_MELLANOX_ARBEL),
1223           .driver_data = ARBEL_NATIVE },
1224         { PCI_DEVICE(PCI_VENDOR_ID_TOPSPIN, PCI_DEVICE_ID_MELLANOX_ARBEL),
1225           .driver_data = ARBEL_NATIVE },
1226         { PCI_DEVICE(PCI_VENDOR_ID_MELLANOX, PCI_DEVICE_ID_MELLANOX_SINAI),
1227           .driver_data = SINAI },
1228         { PCI_DEVICE(PCI_VENDOR_ID_TOPSPIN, PCI_DEVICE_ID_MELLANOX_SINAI),
1229           .driver_data = SINAI },
1230         { PCI_DEVICE(PCI_VENDOR_ID_MELLANOX, PCI_DEVICE_ID_MELLANOX_SINAI_OLD),
1231           .driver_data = SINAI },
1232         { PCI_DEVICE(PCI_VENDOR_ID_TOPSPIN, PCI_DEVICE_ID_MELLANOX_SINAI_OLD),
1233           .driver_data = SINAI },
1234         { 0, }
1235 };
1236
1237 MODULE_DEVICE_TABLE(pci, mthca_pci_table);
1238
1239 static struct pci_driver mthca_driver = {
1240         .name           = DRV_NAME,
1241         .id_table       = mthca_pci_table,
1242         .probe          = mthca_init_one,
1243         .remove         = __devexit_p(mthca_remove_one)
1244 };
1245
1246 static int __init mthca_init(void)
1247 {
1248         int ret;
1249
1250         ret = pci_register_driver(&mthca_driver);
1251         return ret < 0 ? ret : 0;
1252 }
1253
1254 static void __exit mthca_cleanup(void)
1255 {
1256         pci_unregister_driver(&mthca_driver);
1257 }
1258
1259 module_init(mthca_init);
1260 module_exit(mthca_cleanup);