miopen.py

from .env import check_env_flag
from .rocm import USE_ROCM, ROCM_HOME


USE_MIOPEN = False
MIOPEN_LIB_DIR = None
MIOPEN_INCLUDE_DIR = None
MIOPEN_LIBRARY = None
if USE_ROCM and not check_env_flag('NO_MIOPEN'):
    USE_MIOPEN = True
    MIOPEN_LIB_DIR = ROCM_HOME + "/miopen/lib"
    MIOPEN_INCLUDE_DIR = ROCM_HOME + "/miopen/include/miopen"
    MIOPEN_LIBRARY = "MIOpen"
    MIOPEN_FOUND = "True"