/*
 * Copyright (c) 2017-2019, NVIDIA CORPORATION.  All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
// #include <unistd.h>

// const char service_interp[] __attribute__((section(".interp"))) = "/lib/ld-linux.so.2";

#if defined(TARGET_LINUX_X8664) || defined(TARGET_OSX_X8664) || defined(TARGET_WIN_X8664)
#include "cpuid8664.h"
#endif

#ifndef	CPUIDX8664
#define CPUIDX8664(a)  __cpuid_##a
#endif // #ifndef CPUIDX8664

static	int CPUIDX8664(is_avx512)();
static	int CPUIDX8664(is_avx512vl)();
static	int CPUIDX8664(is_avx512f)();
static	int CPUIDX8664(is_avx2)();
static	int CPUIDX8664(is_avx)();
static	int CPUIDX8664(is_intel)();
static	int CPUIDX8664(is_amd)();
static	int CPUIDX8664(is_fma4)();
static	int CPUIDX8664(is_sse4a)();
static	int CPUIDX8664(is_sse41)();

extern const char *get_arch();

const char *get_arch() {
  char *ret = (char *) malloc(sizeof(char) * 10);
  strcpy(ret, "unknown");

  if (CPUIDX8664(is_avx512vl)() == 1) {
    strcpy(ret, "avx512");
  } else if (CPUIDX8664(is_avx512f)() == 1) {
    strcpy(ret, "avx512knl");
  } else if (CPUIDX8664(is_avx2)() == 1) {
    strcpy(ret, "avx2");
  } else if (CPUIDX8664(is_avx)() == 1) {
    if (CPUIDX8664(is_intel)() == 1) {
      strcpy(ret, "avx");
    }
    if (CPUIDX8664(is_amd)() == 1) {
      if (CPUIDX8664(is_fma4)() == 1) {
        strcpy(ret, "avxfma4");
      } else {
        strcpy(ret, "sse4");
      }
    }
  } else {
    if ((CPUIDX8664(is_sse4a)() == 1) || (CPUIDX8664(is_sse41)() == 1)) {
      strcpy(ret, "sse4");
    } else {
      strcpy(ret, "em64t");
    }
  }

  return ret;
}

// void lib_entry(int argc, char **argv) {
//   printf("Instruction Set: %s: ", get_arch());

//   _exit(0);
// }