TheRock/rocm-systems/emulation/rocjitsu/tests/kernels/multikernel_indirect_branch_part1.hip

// Copyright (c) 2026 Advanced Micro Devices, Inc.
// SPDX-License-Identifier: MIT

#include "multikernel_indirect_branch_common.h"

/// Second half of the multi-kernel fixture.  Keeping the same kernel bodies in
/// a smaller code object lets the CFG test compare per-kernel reachability
/// without requiring seven individual fixture builds.
extern "C" __global__ void multikernel_indirect_branch_d(const unsigned *__restrict__ input,
                                                         unsigned *__restrict__ output,
                                                         unsigned n) {
  const unsigned idx = blockIdx.x * blockDim.x + threadIdx.x;
  if (idx >= n)
    return;

  const unsigned base = input[idx] + (idx ^ 0x5a5a5a5au);
  output[idx] = multikernel_indirect_branch_shared_block(base, 67u);
}

extern "C" __global__ void multikernel_indirect_branch_e(const unsigned *__restrict__ input,
                                                         unsigned *__restrict__ output,
                                                         unsigned n) {
  const unsigned idx = blockIdx.x * blockDim.x + threadIdx.x;
  if (idx >= n)
    return;

  const unsigned base = input[idx] ^ (idx + 0x10203040u);
  output[idx] = multikernel_indirect_branch_shared_block(base, 79u);
}

extern "C" __global__ void multikernel_indirect_branch_setpc(const unsigned *__restrict__ input,
                                                             unsigned *__restrict__ output,
                                                             unsigned n) {
  const unsigned idx = blockIdx.x * blockDim.x + threadIdx.x;
  if (idx >= n)
    return;

  const unsigned value = input[idx] + idx;
  if (value == 0x31415926u)
    RJ_STATIC_SETPC_ISLAND(4, 5);
  if (value == 0x27182818u)
    RJ_STATIC_SETPC_ISLAND(6, 7);
  if (value == 0x16180339u)
    RJ_STATIC_SETPC_ISLAND(8, 9);

  output[idx] = value ^ 0xa5a5a5a5u;
}

extern "C" __global__ void multikernel_indirect_branch_scall(const unsigned *__restrict__ input,
                                                             unsigned *__restrict__ output,
                                                             unsigned n) {
  const unsigned idx = blockIdx.x * blockDim.x + threadIdx.x;
  if (idx >= n)
    return;

  const unsigned value = input[idx] ^ (idx * 11u);
  if (value == 0x0badc0deu)
    RJ_STATIC_SCALL_ISLAND(20, 21);
  if (value == 0x00c0ffeeu)
    RJ_STATIC_SCALL_ISLAND(22, 23);
  if (value == 0x1234abcdu)
    RJ_STATIC_SCALL_ISLAND(24, 25);

  output[idx] = value + 1u;
}