static void mlx5_sf_table_enable(struct mlx5_sf_table *table)
 {
-       if (!mlx5_sf_max_functions(table->dev))
-               return;
-
        init_completion(&table->disable_complete);
        refcount_set(&table->refcount, 1);
 }
 
 static void mlx5_sf_table_disable(struct mlx5_sf_table *table)
 {
-       if (!mlx5_sf_max_functions(table->dev))
-               return;
-
        if (!refcount_read(&table->refcount))
                return;
 
 
 static bool mlx5_sf_table_supported(const struct mlx5_core_dev *dev)
 {
-       return dev->priv.eswitch && MLX5_ESWITCH_MANAGER(dev) && mlx5_sf_supported(dev);
+       return dev->priv.eswitch && MLX5_ESWITCH_MANAGER(dev) &&
+              mlx5_sf_hw_table_supported(dev);
 }
 
 int mlx5_sf_table_init(struct mlx5_core_dev *dev)
 
        int err;
        int i;
 
-       if (!table->max_local_functions)
+       if (!table || !table->max_local_functions)
                return -EOPNOTSUPP;
 
        mutex_lock(&table->table_lock);
        /* Dealloc SFs whose firmware event has been missed. */
        mlx5_sf_hw_dealloc_all(table);
 }
+
+bool mlx5_sf_hw_table_supported(const struct mlx5_core_dev *dev)
+{
+       return !!dev->priv.sf_hw_table;
+}
 
 int mlx5_sf_hw_table_sf_alloc(struct mlx5_core_dev *dev, u32 usr_sfnum);
 void mlx5_sf_hw_table_sf_free(struct mlx5_core_dev *dev, u16 id);
 void mlx5_sf_hw_table_sf_deferred_free(struct mlx5_core_dev *dev, u16 id);
+bool mlx5_sf_hw_table_supported(const struct mlx5_core_dev *dev);
 
 #endif