/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 * All rights reserved.
 *
 * This source code is licensed under the BSD-style license found in the
 * LICENSE file in the root directory of this source tree.
 */

#pragma once

#include <array>
#include <string>
#include <set>

namespace libkineto {

// Note : All activity types are not enabled by default. Please add them
// at correct position in the enum
enum class ActivityType {
    // Activity types enabled by default
    CPU_OP = 0, // cpu side ops
    USER_ANNOTATION,
    GPU_USER_ANNOTATION,
    GPU_MEMCPY,
    GPU_MEMSET,
    CONCURRENT_KERNEL, // on-device kernels
    EXTERNAL_CORRELATION,
    CUDA_RUNTIME, // host side cuda runtime events
    CUDA_DRIVER, // host side cuda driver events
    CPU_INSTANT_EVENT, // host side point-like events
    PYTHON_FUNCTION,
    OVERHEAD, // CUPTI induced overhead events sampled from its overhead API.
    MTIA_RUNTIME, // host side MTIA runtime events
    MTIA_CCP_EVENTS, // MTIA ondevice CCP events
    CUDA_SYNC, // synchronization events between runtime and kernels

    // Optional Activity types
    GLOW_RUNTIME, // host side glow runtime events
    CUDA_PROFILER_RANGE, // CUPTI Profiler range for performance metrics
    HPU_OP, // HPU host side runtime event
    XPU_RUNTIME, // host side xpu runtime events
    COLLECTIVE_COMM, // collective communication
    MTIA_WORKLOADD, // MTIA workloadd events

    // PRIVATEUSE1 Activity types are used for custom backends.
    // The corresponding device type is `DeviceType::PrivateUse1` in PyTorch.
    PRIVATEUSE1_RUNTIME, // host side privateUse1 runtime events
    PRIVATEUSE1_DRIVER, // host side privateUse1 driver events

    ENUM_COUNT, // This is to add buffer and not used for any profiling logic. Add your new type before it.
    OPTIONAL_ACTIVITY_TYPE_START = GLOW_RUNTIME,
};

const char* toString(ActivityType t);
ActivityType toActivityType(const std::string& str);

// Return an array of all activity types except COUNT
constexpr int activityTypeCount = (int)ActivityType::ENUM_COUNT;
constexpr int defaultActivityTypeCount = (int)ActivityType::OPTIONAL_ACTIVITY_TYPE_START;
const std::array<ActivityType, activityTypeCount> activityTypes();
const std::array<ActivityType, defaultActivityTypeCount> defaultActivityTypes();

} // namespace libkineto