Source code for aztk.spark.models.plugins.nvblas.configuration
import os
from aztk.models.plugins.plugin_configuration import PluginConfiguration, PluginPort, PluginTargetRole
from aztk.models.plugins.plugin_file import PluginFile
from aztk.utils import constants
dir_path = os.path.dirname(os.path.realpath(__file__))
[docs]def NvBLASPlugin():
return PluginConfiguration(
name="nvblas",
ports=[],
target_role=PluginTargetRole.All,
execute="nvblas.sh",
files=[
PluginFile("nvblas.sh", os.path.join(dir_path, "nvblas.sh")),
]
)