code: def test_dropout_cpu():
      triton_dropout = FusedDropoutBias(p=0.1, bias_shape=None)
      x = torch.normal(0, 1, size=(16, 16), device="cpu")
      _ = triton_dropout(x)
  
      # Check eval means no dropout
      triton_dropout.eval()
      y = triton_dropout(x)
      assert y.count_nonzero() == y.numel()
  
      triton_dropout.train()
      y = triton_dropout(x)
      assert y.count_nonzero() != y.numel()
  
file_path: xformers/tests/test_triton_dropout.py
imports: import logging
  import pytest
  import torch
  from torch.cuda.amp.autocast_mode import autocast
  import xformers
  from xformers.components import Activation, build_activation
label_desc: Use image node_id 1 for calling a global function with example usage: test_dropout_cpu() without return types