{"record":{"id":"12b92596b20549bf","repo":"keras-team/keras","slug":"require-at-least-ampere-arch-to-run-12b925","errorCode":null,"errorMessage":"Require at least Ampere arch to run","messagePattern":"Require at least Ampere arch to run","errorType":"error_code","errorClass":"RuntimeError","httpStatus":null,"severity":"error","filePath":"keras/src/backend/jax/ops/nn.py","lineNumber":1488,"sourceCode":"        from jax.nn import dot_product_attention as dot_product_attention\n    except ImportError:\n        if raise_error:\n            raise ImportError(\n                \"Flash attention is not supported in your current JAX version. \"\n                \"Please update it by following the official guide: \"\n                \"https://jax.readthedocs.io/en/latest/installation.html\"\n            )\n        return False\n\n    if jax.devices()[0].platform == \"tpu\":\n        return True\n    try:\n        # Check if cuDNN is installed and raise RuntimeError if cuDNN is not\n        # detected\n        cudnn_version = check_cudnn_version()\n        # Only support at least Ampere\n        if not check_compute_capability(\"8.0\"):\n            raise RuntimeError(\"Require at least Ampere arch to run\")\n\n        # Inspect inputs of `check_layout`\n        check_layout_params = list(\n            inspect.signature(check_layout).parameters.keys()\n        )\n        for known_param in (\"query\", \"key\", \"value\", \"bias\", \"layout\"):\n            check_layout_params.remove(known_param)\n        # Defaults to `None` when not specified.\n        check_layout_kwargs = {key: None for key in check_layout_params}\n        check_layout(\n            query,\n            key,\n            value,\n            bias,\n            layout=_normalize_layout(\"BTNH\"),\n            **check_layout_kwargs,\n        )\n","sourceCodeStart":1470,"sourceCodeEnd":1506,"githubUrl":"https://github.com/keras-team/keras/blob/7a34a03db60bf60042242d6a556fc3be119046a5/keras/src/backend/jax/ops/nn.py#L1470-L1506","documentation":"Error \"Require at least Ampere arch to run\" thrown in keras-team/keras.","triggerScenarios":"Thrown at keras/src/backend/jax/ops/nn.py:1488 when the library encounters an invalid state.","commonSituations":"See trigger scenarios.","solutions":[],"exampleFix":null,"handlingStrategy":null,"validationCode":null,"typeGuard":null,"tryCatchPattern":null,"preventionTips":[],"tags":[],"backgroundTag":null,"analyzedSha":"7a34a03db60bf60042242d6a556fc3be119046a5","analyzedAt":"2026-08-25T21:25:25.994Z","schemaVersion":2},"datasetVersion":"2026-08-26T02:17:13.382Z"}