TheRock/debug-tools/rocgdb/source/gdb/testsuite/gdb.rocm/hip-builtin-variables.cpp

/* Copyright (C) 2025 Free Software Foundation, Inc.
   Copyright (C) 2025 Advanced Micro Devices, Inc. All rights reserved.

   This file is part of GDB.

   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 3 of the License, or
   (at your option) any later version.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program.  If not, see <http://www.gnu.org/licenses/>.  */

#include <stdio.h>
#include <hip/hip_runtime.h>

#define CHECK(cmd)					\
  {							\
    hipError_t error = cmd;				\
    if (error != hipSuccess)				\
      {							\
	fprintf (stderr, "error: '%s'(%d) at %s:%d\n",	\
		 hipGetErrorString (error), error,	\
		 __FILE__, __LINE__);			\
	exit (EXIT_FAILURE);				\
      }							\
  }

/* A function to be called by "kern ()", without any references to
   threadIdx, blockIdx, blockDim, gridDim, or warpSize variables.  */

__device__ void __attribute__ ((optnone))
inner_func1 ()
{
  return;
}

/* A function to be called by "kern ()", with its own "blockIdx".  */

__device__ void __attribute__ ((optnone))
inner_func2 ()
{
  int blockIdx = 0x42424242;
  return;
}

__global__ void __attribute__ ((optnone))
kern ()
{
  const int thread_idx_x = threadIdx.x;
  const int thread_idx_y = threadIdx.y;
  const int thread_idx_z = threadIdx.z;

  const int group_idx_x = blockIdx.x;
  const int group_idx_y = blockIdx.y;
  const int group_idx_z = blockIdx.z;

  const int group_size_x = blockDim.x;
  const int group_size_y = blockDim.y;
  const int group_size_z = blockDim.z;

  const int grid_size_x = gridDim.x;
  const int grid_size_y = gridDim.y;
  const int grid_size_z = gridDim.z;

  const int wave_size = warpSize;

  /* Break here.  */
  inner_func1 ();

  /* A function with its own local "blockIdx".  */
  inner_func2 ();

  /* This one is supposed to be called after "nosharedlibrary".  */
  inner_func1 ();
}

int
main ()
{
  /* Each workgroup has 64 workitems, so exactly 1 or 2 waves per workgroup
     depending on the architecture.  */
  kern<<<dim3 (3, 4, 5), dim3 (8, 4, 2)>>> ();
  CHECK (hipDeviceSynchronize ());
}