cudammx.py

Go to the documentation of this file.

 
 
 
 
 
from importlib.util import spec_from_loader, module_from_spec
from importlib.machinery import SourceFileLoader, ExtensionFileLoader
import os
import sys
from mm import ModelMatrix
 
ext = "so"
if os.name == 'nt':
    ext="pyd"
try:
    spec = spec_from_loader("cudamarkopy", ExtensionFileLoader("cudamarkopy", os.path.abspath(f"cudamarkopy.{ext}")))
    cudamarkopy = module_from_spec(spec)
    spec.loader.exec_module(cudamarkopy)
except ImportError as e:
    #print(f"({__file__}) Working in development mode. Trying to load cudamarkopy.{ext} from ../../../out/")
    if(os.path.exists(f"../../../out/lib/cudamarkopy.{ext}")):
        spec = spec_from_loader("cudamarkopy", ExtensionFileLoader("cudamarkopy", os.path.abspath(f"../../../out/lib/cudamarkopy.{ext}")))
        cudamarkopy = module_from_spec(spec)
        spec.loader.exec_module(cudamarkopy)
    else:
        raise e
 
 
 
try:
    spec = spec_from_loader("markopy", SourceFileLoader("markopy", os.path.abspath("markopy.py")))
    markopy = module_from_spec(spec)
 
    from mmx import ModelMatrixCLI
    from base import BaseCLI,AbstractGenerationModelCLI, AbstractTrainingModelCLI
 
except ImportError as e:
    #print("Working in development mode. Trying to load from ../../../out/")
    if(os.path.exists("../../../Markopy/src/CLI/markopy.py")):
        spec = spec_from_loader("markopy", SourceFileLoader("markopy", os.path.abspath("../../../Markopy/src/CLI/markopy.py")))
        markopy = module_from_spec(spec)
        sys.path.insert(1, '../../../Markopy/src/CLI/')
 
        from mmx import ModelMatrixCLI
        from base import BaseCLI,AbstractGenerationModelCLI, AbstractTrainingModelCLI
    else:
        raise e
 
import os
import allogate as logging
 
class CudaModelMatrixCLI(ModelMatrixCLI,AbstractGenerationModelCLI):
    """!
         @belongsto Python::CudaMarkopy
         @brief Python CLI wrapper for CudaModelMatrix
         @extends Python::Markopy::ModelMatrixCLI
         @extends Python::Markopy::AbstractGenerationModelCLI
         @extends Markov::API::CUDA::CUDAModelMatrix
    """
    def __init__(self):
        super().__init__()
        self.modelmodelmodelmodel = cudamarkopy.CUDAModelMatrix()
 
    def add_arguments(self):
        super().add_arguments()
        self.parserparser.add_argument("-if", "--infinite", action="store_true", help="Infinite generation mode")
 
    def init_post_arguments(self):
        super().init_post_arguments()
        self.bInfinitebInfinite = self.argsargs.infinite
        
    def _generate(self, wordlist : str ):
        self.modelmodelmodelmodel.FastRandomWalk(int(self.argsargs.count), wordlist, int(self.argsargs.min), int(self.argsargs.max), self.fileIOfileIO, self.bInfinitebInfinite)
 
if __name__ == "__main__":
    mp = CudaModelMatrixCLI()
    mp.parse()
    mp.process()