code: def test_cuda(model, subsample_size):
      tr = poutine.trace(model).get_trace(subsample_size)
      assert tr.log_prob_sum().is_cuda
  
file_path: pyro/tests/poutine/test_mapdata.py
imports: import logging
  import pytest
  import torch
  import pyro
  import pyro.distributions
  import pyro.poutine
  from tests.common import requires_cuda
label_desc: Use image node_id 12 for calling a global function with example usage: test_cuda(model, subsample_size) without return types