code: def test_local_2d_gaussian_distribution(H, W, sigma):
      d = AP.local_2d_gausian_distribution(H, W, sigma=sigma)
      d_ref = _local_2d_gaussian_distribution(H, W, sigma=sigma)
      assert torch.allclose(d, d_ref)
  
imports: import itertools
  import pytest
  import torch
  import xformers.components.attention.attention_patterns
  from xformers.components.attention.sparsity_config import BigBirdSparsityConfig, BSLongformerSparsityConfig, DenseSparsityConfig, FixedSparsityConfig, VariableSparsityConfig
label_desc: Use image node_id 11 for calling a global function with example usage: test_local_2d_gaussian_distribution(H, W, sigma) without return types