Source code for sas.qtgui.Perspectives.Fitting.GPUOptions

# global
import os
import sys
import json
import platform
import webbrowser
import logging
from twisted.internet import threads
from twisted.internet import reactor

import sasmodels
import sasmodels.model_test
import sasmodels.kernelcl
from sasmodels.generate import F32, F64

import sas.qtgui.Utilities.GuiUtils as GuiUtils
from PyQt5 import QtGui, QtCore, QtWidgets
from sas.qtgui.Perspectives.Fitting.UI.GPUOptionsUI import Ui_GPUOptions
from sas.qtgui.Perspectives.Fitting.UI.GPUTestResultsUI import Ui_GPUTestResults

try:
    _fromUtf8 = QtCore.QString.fromUtf8
except AttributeError:
    def _fromUtf8(s):
        return s

try:
    _encoding = QtWidgets.QApplication.UnicodeUTF8
    def _translate(context, text, disambig):
        return QtWidgets.QApplication.translate(context, text, disambig, _encoding)
except AttributeError:
    def _translate(context, text, disambig):
        return QtWidgets.QApplication.translate(context, text, disambig)

logger = logging.getLogger(__name__)

[docs]class GPUOptions(QtWidgets.QDialog, Ui_GPUOptions): """ OpenCL Dialog to select the desired OpenCL driver """ clicked = False sas_open_cl = None cl_options = None testingDoneSignal = QtCore.pyqtSignal(str) testingFailedSignal = QtCore.pyqtSignal(str) def __init__(self, parent=None): super(GPUOptions, self).__init__(parent) self.parent = parent self.setupUi(self) # disable the context help icon self.setWindowFlags(self.windowFlags() & ~QtCore.Qt.WindowContextHelpButtonHint) self.addOpenCLOptions() self.progressBar.setVisible(False) self.progressBar.setFormat(" Test %v / %m") self.createLinks() self.testingDoneSignal.connect(self.testCompleted) self.testingFailedSignal.connect(self.testFailed)
[docs] def addOpenCLOptions(self): """ Populate the window with a list of OpenCL options """ # Get list of openCL options and add to GUI cl_tuple = _get_clinfo() self.sas_open_cl = os.environ.get("SAS_OPENCL", "") # Keys are the names in the form "platform: device". Corresponding values are the combined indices, e.g. # "0:1", for setting the SAS_OPENCL env. self.cl_options = {} for title, descr in cl_tuple: # Create an item for each openCL option check_box = QtWidgets.QCheckBox() check_box.setObjectName(_fromUtf8(descr)) check_box.setText(_translate("GPUOptions", descr, None)) self.optionsLayout.addWidget(check_box) if (title == self.sas_open_cl) or ( title == "None" and not self.clicked): check_box.click() self.clicked = True self.cl_options[descr] = title self.openCLCheckBoxGroup.setMinimumWidth(self.optionsLayout.sizeHint().width()+10)
[docs] def checked(self): """ Only allow a single check box to be selected. Uncheck others. """ checked = None for box in self.openCLCheckBoxGroup.findChildren(QtWidgets.QCheckBox): if box.isChecked() and (self.cl_options[str(box.text())] == self.sas_open_cl or ( str(box.text()) == "No OpenCL" and self.sas_open_cl == "")): box.setChecked(False) elif box.isChecked(): checked = box if hasattr(checked, "text"): self.sas_open_cl = self.cl_options[str(checked.text())] else: self.sas_open_cl = None
[docs] def set_sas_open_cl(self): """ Set SAS_OPENCL value when tests run or OK button clicked """ no_opencl_msg = False if self.sas_open_cl: os.environ["SAS_OPENCL"] = self.sas_open_cl if self.sas_open_cl.lower() == "none": no_opencl_msg = True else: if "SAS_OPENCL" in os.environ: del os.environ["SAS_OPENCL"] # CRUFT: next version of reset_environment() will return env sasmodels.sasview_model.reset_environment() return no_opencl_msg
[docs] def testButtonClicked(self): """ Run sasmodels check from here and report results from """ self.model_tests = sasmodels.model_test.make_suite('opencl', ['all']) number_of_tests = len(self.model_tests._tests) self.progressBar.setMinimum(0) self.progressBar.setMaximum(number_of_tests) self.progressBar.setVisible(True) self.testButton.setEnabled(False) self.okButton.setEnabled(False) self.resetButton.setEnabled(False) no_opencl_msg = self.set_sas_open_cl() test_thread = threads.deferToThread(self.testThread, no_opencl_msg) test_thread.addCallback(self.testComplete) test_thread.addErrback(self.testFail)
[docs] def testThread(self, no_opencl_msg): """ Testing in another thread """ try: env = sasmodels.kernelcl.environment() clinfo = {} if env.context[F64] is None: clinfo['double'] = "None" else: ctx64 = env.context[F64].devices[0] clinfo['double'] = ", ".join(( ctx64.platform.vendor, ctx64.platform.version, ctx64.vendor, ctx64.name, ctx64.version)) if env.context[F32] is None: clinfo['single'] = "None" else: ctx32 = env.context[F32].devices[0] clinfo['single'] = ", ".join(( ctx32.platform.vendor, ctx32.platform.version, ctx32.vendor, ctx32.name, ctx32.version)) # If the same device is used for single and double precision, then # say so. Whether double is the same as single or single is the # same as double depends on the order they are listed below. if env.context[F32] == env.context[F64]: clinfo['double'] = "same as single precision" except Exception as exc: logger.debug("exc %s", str(exc)) clinfo = {'double': "None", 'single': "None"} failures = [] tests_completed = 0 for counter, test in enumerate(self.model_tests): # update the progress bar reactor.callFromThread(self.updateCounter, counter) try: test.run_all() except Exception: failures.append(test.test_name) tests_completed += 1 info = { 'version': sasmodels.__version__, 'platform': platform.uname(), 'opencl_single': clinfo['single'], 'opencl_double': clinfo['double'], 'failing tests': failures, } msg = str(tests_completed) + ' tests completed.\n' if len(failures) > 0: msg += str(len(failures)) + ' tests failed.\n' msg += 'Failing tests: ' msg += json.dumps(info['failing tests']) msg += "\n" else: msg += "All tests passed!\n" msg += "\nPlatform Details:\n\n" msg += "Sasmodels version: " msg += info['version'] + "\n" msg += "\nPlatform used: " msg += json.dumps(info['platform']) + "\n" if no_opencl_msg: msg += "\nOpenCL driver: None" else: msg += "\nOpenCL driver: " msg += " single precision: " + json.dumps(info['opencl_single']) + "\n" msg += " double precision: " + json.dumps(info['opencl_double']) + "\n" return msg
[docs] def updateCounter(self, step): """ Update progress bar with current value """ self.progressBar.setValue(step) return
[docs] def testComplete(self, msg): """ Testing done: send signal to main thread with update """ self.testingDoneSignal.emit(msg)
[docs] def testFail(self, msg): """ Testing failed: log the reason """ from twisted.python.failure import Failure if isinstance(msg, Failure): msg = msg.getErrorMessage() self.testingFailedSignal.emit(msg)
[docs] def testFailed(self, msg): """ Testing failed: log the reason """ self.progressBar.setVisible(False) self.testButton.setEnabled(True) self.okButton.setEnabled(True) self.resetButton.setEnabled(True) logging.error(str(msg))
[docs] def testCompleted(self, msg): """ Respond to successful test completion """ self.progressBar.setVisible(False) self.testButton.setEnabled(True) self.okButton.setEnabled(True) self.resetButton.setEnabled(True) GPUTestResults(self, msg)
[docs] def helpButtonClicked(self): """ Open the help menu when the help button is clicked """ help_location = GuiUtils.HELP_DIRECTORY_LOCATION help_location += "/user/qtgui/Perspectives/Fitting/gpu_setup.html" help_location += "#device-selection" # Display the page in default browser webbrowser.open('file://' + os.path.realpath(help_location))
[docs] def reject(self): """ Close the window without modifying SAS_OPENCL """ self.closeEvent(None) self.parent.gpu_options_widget.open()
[docs] def accept(self): """ Close the window after modifying the SAS_OPENCL value """ self.set_sas_open_cl() self.closeEvent(None)
[docs] def closeEvent(self, event): """ Overwrite QDialog close method to allow for custom widget close """ self.close() self.parent.gpu_options_widget = GPUOptions(self.parent)
[docs]class GPUTestResults(QtWidgets.QDialog, Ui_GPUTestResults): """ OpenCL Dialog to modify the OpenCL options """ def __init__(self, parent, msg): super(GPUTestResults, self).__init__(parent) self.setupUi(self) self.resultsText.setText(_translate("GPUTestResults", msg, None)) #self.setFixedSize(self.size()) self.open()
def _get_clinfo(): """ Read in information about available OpenCL infrastructure """ clinfo = [] cl_platforms = [] try: import pyopencl as cl except ImportError: cl = None if cl is None: logger.warn("Unable to import the pyopencl package. It may not " "have been installed. If you wish to use OpenCL, try " "running pip install --user pyopencl") else: try: cl_platforms = cl.get_platforms() except cl.LogicError as err: logger.warn("Unable to fetch the OpenCL platforms. This likely " "means that the opencl drivers for your system are " "not installed.") logger.warn(err) p_index = 0 for cl_platform in cl_platforms: d_index = 0 cl_devices = cl_platform.get_devices() for cl_device in cl_devices: if len(cl_platforms) > 1 and len(cl_devices) > 1: combined_index = ":".join([str(p_index), str(d_index)]) elif len(cl_platforms) > 1: combined_index = str(p_index) else: combined_index = str(d_index) clinfo.append((combined_index, ": ".join([cl_platform.name, cl_device.name]))) d_index += 1 p_index += 1 clinfo.append(("None", "No OpenCL")) return clinfo