code: def test_cpu_forward(self):
      input_ids = torch.tensor([0, 1, 2, 3], dtype=torch.int64).repeat(
          (2, 1)
      )
      position_ids = torch.tensor(
          [0, 1, 2, 3], dtype=torch.int64
      ).repeat((2, 1))
      embeddings = self.base_embedding(input_ids, position_ids)
      assert embeddings.device.type == "cpu"
      assert (
          embeddings.shape[0] == self.base_embedding.max_sequence_length
      )
      assert embeddings.shape[1] == input_ids.shape[0]
      assert (
          embeddings.shape[2] == self.base_embedding.config.hidden_size
      )
  
file_path: megatron-lm/tests/unit_tests/models/test_base_embedding.py
imports: import pytest
  import torch
  from megatron.core.transformer.transformer_config import TransformerConfig
  from megatron.core.models.common.embeddings.language_model_embedding import LanguageModelEmbedding
  from tests.unit_tests.test_utilities import Utils
label_desc: Use image node_id 5 for calling the TestBaseEmbedding obj's underlying member method code with example usage: obj.test_cpu_forward() without return types