#!/bin/bash
#SBATCH --job-name=cupy_gpu_job
#SBATCH --partition=unite           # Target partition for the UNITE cluster
#SBATCH --output=cupy_job_%j.out
#SBATCH --error=cupy_job_%j.err
#SBATCH --nodes=1                   # Run on a single node
#SBATCH --ntasks=1                  # Run a single task instance
#SBATCH --gres=gpu:1                # Request 1 GPU allocation
#SBATCH --cpus-per-task=4           # Allocate CPU cores for background thread pooling
#SBATCH --time=00:15:00             # Format: HH:MM:SS

# Clean out inherited environment parameters
module purge

# Crucial: Load the exact same Python and CuPy dependency module matrix
module load unite/python/3.14/python-3.14.0
module load unite/python/3.14/cupy

# Launch your Python script
python3 cupy_test.py