Hugging Face's logo

Spaces:
kernels-community
/
kernels-benchmarks
Running

App Files Community
kernels-benchmarks / openai_moe /impls /gpt_oss_moe.html
drbh's picture
drbh HF Staff
Upload folder using huggingface_hub
9b13459 verified
raw
history blame
198 kB
 <!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>gpt_oss_moe</title>
<link rel="preconnect" href="https://fonts.googleapis.com">
<link rel="preconnect" href="https://fonts.gstatic.com" crossorigin>
<link href="https://fonts.googleapis.com/css2?family=JetBrains+Mono:ital,wght@0,100..800;1,100..800&display=swap" rel="stylesheet">
<script>
// Iframe-friendly navigation router
(function() {
const isIframe = window.self !== window.top;
if (!isIframe) return; // Only activate in iframe context
// On load: if hash points to a different page, navigate there
const hash = window.location.hash;
if (hash && hash.startsWith('#/')) {
const targetPath = hash.slice(2); // Remove '#/'
const currentPath = window.location.pathname.split('/').pop();
// Only navigate if we're not already on the target page
if (targetPath !== currentPath) {
window.location.href = targetPath;
return; // Stop execution, we're navigating away
}
}
// Intercept all link clicks for hash-based navigation
document.addEventListener('click', function(e) {
const link = e.target.closest('a');
if (!link) return;
const href = link.getAttribute('href');
// Skip external links, anchors, and javascript: links
if (!href || href.startsWith('#') || href.startsWith('http') || href.startsWith('javascript:')) {
return;
}
e.preventDefault();
// Convert relative/absolute path to hash-based navigation
const url = new URL(href, window.location.href);
let fullPath = url.pathname;
// Remove leading slash if present for cleaner paths
if (fullPath.startsWith('/')) {
fullPath = fullPath.slice(1);
}
// Update parent URL hash
window.location.hash = '#/' + fullPath;
// For HTML files, navigate within iframe
if (fullPath.endsWith('.html') || fullPath.endsWith('/')) {
const pathParts = fullPath.split('/').filter(p => p);
const targetFile = pathParts[pathParts.length - 1] || 'index.html';
window.location.href = targetFile;
} else {
// For non-HTML files (raw .py, etc), open directly
window.open(href, '_blank');
}
});
})();
// Apply theme and widget visibility immediately to prevent flicker
(function() {
const configTheme = 'dark';
const hasConfigUi = false;
const configUi = hasConfigUi ? 'None' : null;
const hasWidgetsConfig = false;
const widgetsOn = hasWidgetsConfig ? false : true;
let theme;
if (configTheme === 'auto') {
theme = window.matchMedia('(prefers-color-scheme: dark)').matches ? 'dark' : 'light';
} else {
theme = localStorage.getItem('uvnote-theme') || configTheme;
}
document.documentElement.setAttribute('data-theme', theme);
// Initialize UI theme (css theme)
let ui = hasConfigUi ? configUi : (localStorage.getItem('uvnote-ui') || 'default');
if (ui !== 'default' && ui !== 'none' && ui !== 'monocolor') { ui = 'default'; }
document.documentElement.setAttribute('data-ui', ui);
// Apply widgets visibility
document.documentElement.setAttribute('data-widgets', widgetsOn ? 'on' : 'off');
})();
</script>
<style>
:root[data-theme="light"] {
--bg-primary: #ffffff;
--bg-secondary: #f6f8fa;
--bg-tertiary: #f8f9fa;
--bg-code: #f8f9fa;
--bg-error: #fdf2f2;
--bg-artifact: #e6f3ff;
--bg-artifact-hover: #d0e7ff;
--text-primary: #333;
--text-secondary: #656d76;
--text-error: #c53030;
--text-link: #0969da;
--border-primary: #e1e5e9;
--border-error: #e53e3e;
--border-cell-failed: #d73a49;
--shadow: rgba(0, 0, 0, 0.1);
}
:root[data-theme="dark"] {
--bg-primary: #0a0a0a;
--bg-secondary: #121212;
--bg-tertiary: #181818;
--bg-code: #0d0d0d;
--bg-error: #1a0f0f;
--bg-artifact: #151515;
--bg-artifact-hover: #1a1a1a;
--text-primary: #e0e0e0;
--text-secondary: #888888;
--text-error: #ff6b6b;
--text-link: #64b5f6;
--border-primary: #2a2a2a;
--border-error: #ff6b6b;
--border-cell-failed: #ff6b6b;
--shadow: rgba(255, 255, 255, 0.05);
}
/* Monocolor UI theme: black/white background, all text/borders single blue */
:root[data-ui="monocolor"] {
--mono-color: #0a66ff;
}
:root[data-ui="monocolor"][data-theme="light"] {
--bg-primary: #ffffff;
}
:root[data-ui="monocolor"][data-theme="dark"] {
--bg-primary: #000000;
}
:root[data-ui="monocolor"] {
--bg-secondary: var(--bg-primary);
--bg-tertiary: var(--bg-primary);
--bg-code: var(--bg-primary);
--bg-error: var(--bg-primary);
--bg-artifact: var(--bg-primary);
--bg-artifact-hover: var(--bg-primary);
--text-primary: var(--mono-color);
--text-secondary: var(--mono-color);
--text-error: var(--mono-color);
--text-link: var(--mono-color);
--border-primary: var(--mono-color);
--border-error: var(--mono-color);
--border-cell-failed: var(--mono-color);
--shadow: none;
}
:root[data-ui="monocolor"] a {
color: var(--mono-color);
}
:root[data-ui="monocolor"] .menu-button,
:root[data-ui="monocolor"] .theme-toggle,
:root[data-ui="monocolor"] .reset-toggle,
:root[data-ui="monocolor"] .back-button {
background: var(--bg-primary);
color: var(--mono-color);
border-color: var(--mono-color);
}
:root[data-ui="monocolor"] .menu-button:hover,
:root[data-ui="monocolor"] .theme-toggle:hover,
:root[data-ui="monocolor"] .reset-toggle:hover,
:root[data-ui="monocolor"] .back-button:hover {
background: var(--bg-primary);
color: var(--mono-color);
border-color: var(--mono-color);
}
:root[data-ui="monocolor"] .menu-dropdown {
background: var(--bg-primary);
border-color: var(--mono-color);
box-shadow: none;
}
:root[data-ui="monocolor"] .menu-item {
color: var(--mono-color);
border-bottom-color: var(--mono-color);
}
:root[data-ui="monocolor"] .system-info {
background: var(--bg-primary);
border-color: var(--mono-color);
}
:root[data-ui="monocolor"] .cell {
border-color: var(--mono-color);
background: var(--bg-primary);
}
:root[data-ui="monocolor"] .cell-header {
background: var(--bg-primary);
border-bottom-color: var(--mono-color);
}
:root[data-ui="monocolor"] .artifact {
background: var(--bg-primary);
border-color: var(--mono-color);
color: var(--mono-color);
}
:root[data-ui="monocolor"] .artifact:hover {
background: var(--bg-primary);
}
:root[data-ui="monocolor"] .artifact-preview img,
:root[data-ui="monocolor"] .artifact-preview svg {
border-color: var(--mono-color);
}
:root[data-ui="monocolor"] .status-widget {
background: var(--bg-primary);
border-color: var(--mono-color);
color: var(--mono-color);
}
:root[data-ui="monocolor"] .minimap,
:root[data-ui="monocolor"] .file-explorer,
:root[data-ui="monocolor"] .tools-widget {
background: var(--bg-primary);
border-color: var(--mono-color);
color: var(--mono-color);
}
:root[data-ui="monocolor"] .cell-code {
background: var(--bg-primary);
border-bottom-color: var(--mono-color);
}
:root[data-ui="monocolor"] .tools-title,
:root[data-ui="monocolor"] .file-explorer-section-title,
:root[data-ui="monocolor"] .minimap-title {
color: var(--mono-color);
border-bottom-color: var(--mono-color);
}
:root[data-ui="monocolor"] .tool-button {
background: var(--bg-primary);
border-color: var(--mono-color);
color: var(--mono-color);
}
:root[data-ui="monocolor"] .tool-button.active {
border-color: var(--mono-color);
}
:root[data-ui="monocolor"] .file-explorer-item,
:root[data-ui="monocolor"] .minimap-item {
color: var(--mono-color);
}
/* Force Pygments code to mono blue on mono bg */
:root[data-ui="monocolor"] .highlight {
background: var(--bg-primary) !important;
color: var(--mono-color) !important;
}
:root[data-ui="monocolor"] .highlight *,
:root[data-ui="monocolor"] .highlight .hll {
color: var(--mono-color) !important;
background: transparent !important;
border-color: var(--mono-color) !important;
}
/* Default code font + metrics (overridable via frontmatter) */
:root {
--code-font-size: 0.95rem;
--code-line-height: 1.5;
--code-pad-y: 0.75rem;
}
/* Minimal UI theme overrides base variables for a flatter, 90s look */
:root[data-ui="none"] {
--bg-primary: #ffffff;
--bg-secondary: transparent;
--bg-tertiary: transparent;
--bg-code: #f9f9f9;
--bg-error: #fff0f0;
--bg-artifact: #f0f7ff;
--bg-artifact-hover: #e5f1ff;
--text-primary: #000000;
--text-secondary: #222222;
--text-error: #a00000;
--text-link: #0000ee;
--border-primary: #cccccc;
--border-error: #cc0000;
--border-cell-failed: #cc0000;
--shadow: none;
}
html {
overscroll-behavior: none;
}
body {
font-family: 'Cascadia Mono', 'Cascadia Code', 'JetBrains Mono', 'SF Mono', Monaco, 'Consolas', monospace;
line-height: 1.4;
max-width: 1000px;
margin: 0 auto;
padding: 15px;
color: var(--text-primary);
background: var(--bg-primary);
transition: background-color 0.2s ease, color 0.2s ease;
overscroll-behavior: none;
}
/* Minimal "none" UI theme overrides */
:root[data-ui="none"] body {
font-family: 'Times New Roman', Times, serif;
line-height: 1.5;
max-width: 860px;
padding: 12px;
background: #ffffff;
color: #000000;
transition: none;
}
/* Two panel layout removed */
.controls {
position: fixed;
top: 20px;
right: 20px;
display: flex;
flex-direction: column;
align-items: flex-end;
gap: 0.25rem;
z-index: 1000;
}
.controls-buttons {
display: flex;
gap: 0.5rem;
}
.menu-button {
position: relative;
background: var(--bg-secondary);
border: 1px solid var(--border-primary);
padding: 8px 12px;
border-radius: 2px;
color: var(--text-secondary);
cursor: pointer;
font-family: inherit;
font-size: 0.9rem;
user-select: none;
}
/* Keep default control styling when widgets are enabled, even in minimal UI */
:root[data-ui="none"][data-widgets="on"] .menu-button,
:root[data-ui="none"][data-widgets="on"] .theme-toggle,
:root[data-ui="none"][data-widgets="on"] .reset-toggle,
:root[data-ui="none"][data-widgets="on"] .back-button {
background: #f6f6f6;
border: 1px solid #cccccc;
color: #222222;
}
.menu-button:hover {
color: var(--text-primary);
background: var(--bg-tertiary);
}
/* Controls state indicator (top-right) */
/* Status widget (bottom-right) */
.status-widget {
position: fixed;
right: 20px;
bottom: 20px;
width: auto;
max-width: 260px;
background: var(--bg-secondary);
border: 1px solid var(--border-primary);
border-radius: 2px;
padding: 6px 8px;
font-size: 0.8rem;
color: var(--text-secondary);
z-index: 100;
}
.status-widget strong {
color: var(--text-primary);
}
:root[data-ui="none"][data-widgets="on"] .status-widget {
background: #f6f6f6;
border-color: #ccc;
color: #222;
}
:root[data-ui="none"][data-widgets="on"] .menu-button:hover,
:root[data-ui="none"][data-widgets="on"] .theme-toggle:hover,
:root[data-ui="none"][data-widgets="on"] .reset-toggle:hover,
:root[data-ui="none"][data-widgets="on"] .back-button:hover {
background: #ededed;
border-color: #bbbbbb;
color: #000000;
}
.menu-dropdown {
position: absolute;
top: 100%;
right: 0;
background: var(--bg-secondary);
border: 1px solid var(--border-primary);
border-radius: 4px;
box-shadow: 0 4px 12px var(--shadow);
min-width: 160px;
opacity: 0;
visibility: hidden;
transform: translateY(-8px);
transition: all 0.2s ease;
z-index: 1001;
margin-top: 4px;
}
:root[data-ui="none"][data-widgets="on"] .menu-dropdown {
background: #ffffff;
border: 1px solid #cccccc;
box-shadow: none;
}
.menu-button.active .menu-dropdown {
opacity: 1;
visibility: visible;
transform: translateY(0);
}
.menu-item {
display: block;
padding: 8px 12px;
color: var(--text-secondary);
text-decoration: none;
font-size: 0.85rem;
border-bottom: 1px solid var(--border-primary);
cursor: pointer;
}
:root[data-ui="none"] .menu-item {
color: #000;
border-bottom: 1px solid #eee;
}
.menu-item:last-child {
border-bottom: none;
}
.menu-item:hover {
background: var(--bg-tertiary);
color: var(--text-primary);
}
.menu-checkbox {
display: inline-block;
width: 16px;
font-family: monospace;
color: var(--text-link);
}
.theme-toggle,
.reset-toggle,
.back-button {
background: var(--bg-secondary);
border: 1px solid var(--border-primary);
padding: 8px 12px;
border-radius: 4px;
color: var(--text-secondary);
cursor: pointer;
font-family: inherit;
font-size: 0.9rem;
user-select: none;
}
.back-button {
text-decoration: none;
display: inline-block;
}
.theme-toggle:hover,
.reset-toggle:hover,
.back-button:hover {
color: var(--text-primary);
background: var(--bg-tertiary);
}
.system-info {
background: var(--bg-secondary);
border: 1px solid var(--border-primary);
border-radius: 4px;
padding: 8px 12px;
margin-bottom: 16px;
font-size: 0.85em;
color: var(--text-secondary);
}
.system-info-header {
font-weight: 600;
color: var(--text-primary);
margin-bottom: 2px;
}
.system-info-content {
font-family: monospace;
}
.theme-toggle,
.reset-toggle {
background: var(--bg-secondary);
border: 1px solid var(--border-primary);
border-radius: 2px;
/* padding: 0.4rem 0.6rem; */
cursor: pointer;
font-family: inherit;
font-size: 0.8rem;
color: var(--text-secondary);
user-select: none;
transition: all 0.2s ease;
text-transform: lowercase;
letter-spacing: 0;
}
.theme-toggle:hover,
.reset-toggle:hover {
background: var(--bg-tertiary);
border-color: var(--text-secondary);
color: var(--text-primary);
}
.minimap {
position: fixed;
bottom: 20px;
right: 20px;
width: 220px;
max-height: 400px;
background: var(--bg-secondary);
border: 1px solid var(--border-primary);
border-radius: 2px;
padding: 0.5rem;
font-size: 0.7rem;
overflow-y: auto;
z-index: 100;
opacity: 0.9;
transition: opacity 0.2s ease;
}
/* Hide widgets and controls when disabled via frontmatter */
:root[data-widgets="off"] .controls,
:root[data-widgets="off"] .minimap,
:root[data-widgets="off"] .file-explorer,
:root[data-widgets="off"] .tools-widget,
:root[data-widgets="off"] .status-widget {
display: none !important;
}
.file-explorer {
position: fixed;
bottom: 20px;
/* default; JS will stack */
right: 20px;
left: auto;
top: auto;
width: 220px;
max-height: 400px;
background: var(--bg-secondary);
border: 1px solid var(--border-primary);
border-radius: 2px;
padding: 0.5rem;
font-size: 0.7rem;
overflow-y: auto;
z-index: 100;
opacity: 0.9;
transition: opacity 0.2s ease;
}
/* Drawing overlay */
.draw-overlay {
position: fixed;
top: 0;
left: 0;
width: 100vw;
height: 100vh;
z-index: 80;
/* under widgets (100) and controls (1000) */
display: block;
pointer-events: none;
/* enabled only when a tool is active */
}
/* Tools widget */
.tools-widget {
position: fixed;
bottom: 20px;
/* default; JS will stack */
right: 20px;
left: auto;
top: auto;
width: 220px;
background: var(--bg-secondary);
border: 1px solid var(--border-primary);
border-radius: 2px;
padding: 0.5rem;
font-size: 0.7rem;
z-index: 100;
opacity: 0.95;
}
.tools-title {
font-weight: bold;
color: var(--text-secondary);
margin-bottom: 0.5rem;
padding-bottom: 0.25rem;
border-bottom: 1px solid var(--border-primary);
cursor: grab;
user-select: none;
}
.tools-row {
display: flex;
gap: 0.4rem;
flex-wrap: wrap;
}
.tool-button {
background: var(--bg-tertiary);
border: 1px solid var(--border-primary);
border-radius: 2px;
padding: 0.25rem 0.4rem;
cursor: pointer;
color: var(--text-secondary);
font-family: inherit;
font-size: 0.75rem;
user-select: none;
}
.tool-button:hover {
color: var(--text-primary);
}
.tool-button.active {
color: var(--text-primary);
border-color: var(--text-secondary);
background: var(--bg-secondary);
}
.minimap:hover,
.file-explorer:hover {
opacity: 1;
}
.minimap-title {
font-weight: bold;
color: var(--text-secondary);
margin-bottom: 0.5rem;
padding-bottom: 0.25rem;
border-bottom: 1px solid var(--border-primary);
cursor: grab;
/* drag handle */
user-select: none;
}
.minimap-item {
display: block;
color: var(--text-secondary);
text-decoration: none;
padding: 0.15rem 0;
border-left: 2px solid transparent;
padding-left: 0.5rem;
transition: all 0.2s ease;
cursor: pointer;
}
.minimap-item:hover {
color: var(--text-primary);
border-left-color: var(--text-secondary);
}
.minimap-item.active {
color: var(--text-primary);
border-left-color: var(--text-link);
}
.minimap-heading {
font-weight: normal;
}
.minimap-heading.h1 {
padding-left: 0.5rem;
}
.minimap-heading.h2 {
padding-left: 1rem;
}
.minimap-heading.h3 {
padding-left: 1.5rem;
}
.minimap-heading.h4 {
padding-left: 2rem;
}
.minimap-heading.h5 {
padding-left: 2.5rem;
}
.minimap-heading.h6 {
padding-left: 3rem;
}
.minimap-cell {
color: var(--text-link);
opacity: 0.8;
font-style: italic;
}
.minimap-cell:hover {
opacity: 1;
}
.file-explorer-title {
font-weight: bold;
color: var(--text-secondary);
margin-bottom: 0.5rem;
padding-bottom: 0.25rem;
border-bottom: 1px solid var(--border-primary);
cursor: grab;
/* drag handle */
user-select: none;
}
.file-explorer-section {
margin-bottom: 0.75rem;
}
.file-explorer-section-title {
font-weight: bold;
color: var(--text-secondary);
font-size: 0.65rem;
margin-bottom: 0.25rem;
text-transform: uppercase;
letter-spacing: 0.5px;
}
.file-explorer-item {
display: block;
color: var(--text-secondary);
text-decoration: none;
padding: 0.1rem 0;
margin-left: 0.5rem;
transition: color 0.2s ease;
cursor: pointer;
font-family: monospace;
}
.file-explorer-item:hover {
color: var(--text-primary);
}
.file-explorer-item.script {
color: var(--text-link);
}
.file-explorer-item.artifact {
color: var(--text-secondary);
opacity: 0.8;
}
/* Hide widgets on smaller screens */
@media (max-width: 768px) {
.minimap,
.file-explorer,
.tools-widget {
display: none;
}
}
.cell {
margin: 1rem 0;
border: 1px solid var(--border-primary);
border-radius: 2px;
overflow: hidden;
background: var(--bg-secondary);
}
:root[data-ui="none"] .cell {
margin: 1em 0;
border: none;
background: transparent;
}
.cell-header {
background: var(--bg-secondary);
padding: 0.5rem 1rem;
border-bottom: 1px solid var(--border-primary);
font-family: inherit;
font-size: 0.85rem;
}
:root[data-ui="none"] .cell-header {
background: transparent;
border: none;
padding: 0;
font-weight: bold;
}
:root[data-ui="none"] .cell-content {
padding: 0;
}
:root[data-ui="none"] .copy-button,
:root[data-ui="none"] .collapse-indicators,
:root[data-ui="none"] .cell-meta,
:root[data-ui="none"] .cell-outputs-header {
display: none !important;
}
:root[data-ui="none"] pre,
:root[data-ui="none"] code {
font-family: Menlo, Monaco, 'Courier New', monospace;
}
:root[data-ui="none"] .code-content pre {
background: #f9f9f9;
border: 1px solid #ddd;
padding: 8px;
}
:root[data-ui="none"] .output {
background: transparent;
border: none;
padding: 0.25em 0;
}
color: var(--text-secondary);
cursor: pointer;
user-select: none;
transition: background-color 0.2s ease;
}
.cell-header:hover {
background: var(--bg-tertiary);
}
.collapse-indicators {
color: var(--text-secondary);
font-size: 0.8rem;
opacity: 0.7;
}
.collapse-indicators span:hover {
color: var(--text-primary);
opacity: 1;
}
.cell-code {
display: block;
background: var(--bg-code);
}
.cell-code.collapsed {
display: none;
}
.cell-code pre {
margin: 0;
padding: 0.75rem;
background: var(--bg-code);
overflow-x: auto;
color: var(--text-primary);
}
.cell-output {
padding: 0.75rem;
/* background: var(--bg-primary); */
background: var(--bg-secondary);
}
.cell-output.collapsed {
display: none;
}
.cell-stdout {
background: var(--bg-tertiary);
padding: 0.75rem;
border-radius: 1px;
/* margin: 0.25rem 0; */
font-family: inherit;
font-size: 0.9rem;
white-space: pre-wrap;
color: var(--text-primary);
}
.cell-stdout {
background: var(--bg-tertiary);
padding: 0.75rem;
border-radius: 1px;
font-family: inherit;
font-size: 0.9rem;
color: var(--text-primary);
/* key bits */
overflow: auto;
/* show scrollbars when needed */
max-width: 100%;
/* respects whatever layout width you give it */
}
.cell-stdout .stdout-text {
margin: 0;
/* reset pre default margin */
white-space: pre;
/* keep line breaks, NO wrapping */
display: inline-block;
/* shrink-to-content */
min-width: max-content;
/* allow very long lines to define intrinsic width */
font-family: ui-monospace, SFMono-Regular, Menlo, Consolas, monospace;
tab-size: 2;
}
.cell-stderr {
background: var(--bg-error);
border-left: 2px solid var(--border-error);
padding: 1rem;
margin: 0.5rem 0;
font-family: inherit;
font-size: 0.9rem;
color: var(--text-error);
white-space: pre-wrap;
}
.uv-install-logs {
margin: 0.5rem 0;
}
.uv-logs-header {
cursor: pointer;
padding: 0.75rem;
border-left: 3px solid var(--border-color);
font-family: inherit;
font-size: 0.85rem;
color: var(--text-secondary);
user-select: none;
}
.uv-logs-content {
background: var(--bg-secondary);
padding: 1rem;
border-left: 3px solid var(--border-color);
white-space: pre-wrap;
font-family: monospace;
font-size: 0.85rem;
color: var(--text-secondary);
overflow-x: auto;
}
.cell-artifacts {
margin: 1rem 0;
}
.cell-artifacts h4 {
margin: 0 0 0.5rem 0;
color: var(--text-secondary);
font-size: 0.9rem;
}
.artifact {
display: inline-block;
background: var(--bg-artifact);
padding: 0.25rem 0.5rem;
border-radius: 1px;
margin: 0.25rem 0.5rem 0.25rem 0;
font-family: inherit;
font-size: 0.8rem;
color: var(--text-link);
text-decoration: none;
transition: background-color 0.2s ease;
border: 1px solid var(--border-primary);
}
.artifact:hover {
background: var(--bg-artifact-hover);
}
.artifact-preview {
margin-top: 1rem;
}
.artifact-preview img {
max-width: 100%;
height: auto;
border: 1px solid var(--border-primary);
border-radius: 1px;
}
.artifact-preview svg {
max-width: 100%;
height: auto;
border: 1px solid var(--border-primary);
border-radius: 1px;
display: block;
}
/* Style SVG text elements */
.artifact-preview svg g {
fill: var(--text-primary) !important;
}
/* Auto-theme SVG elements */
.artifact-preview svg {
background: transparent;
}
/* Invert SVG images in dark mode */
:root[data-theme="dark"] .artifact-preview img[src$=".svg"] {
filter: invert(0.9) hue-rotate(180deg);
}
/* Keep SVG images readable in monocolor mode */
:root[data-ui="monocolor"] .artifact-preview img[src$=".svg"] {
filter: none;
}
/* CSV table styling */
.artifact-csv {
margin-top: 1rem;
overflow-x: auto;
}
.csv-table {
width: 100%;
border-collapse: collapse;
font-size: 0.9rem;
background: var(--bg-secondary);
border: 1px solid var(--border-primary);
border-radius: 1px;
}
.csv-table th,
.csv-table td {
padding: 0.5rem 0.75rem;
text-align: left;
border: 1px solid var(--border-primary);
}
.csv-table th {
background: var(--bg-tertiary);
font-weight: 600;
color: var(--text-primary);
}
.csv-table tbody tr:hover {
background: var(--bg-artifact-hover);
}
.artifact-csv-error {
margin-top: 1rem;
padding: 1rem;
background: var(--bg-error);
color: var(--text-error);
border: 1px solid var(--border-error);
border-radius: 1px;
}
.cell-failed {
border-color: var(--border-cell-failed);
}
.cell-failed .cell-header {
background: var(--bg-error);
color: var(--text-error);
}
.cell-commented {
opacity: 0.6;
border-style: dashed;
}
.cell-commented .cell-header {
background: var(--bg-secondary);
color: var(--text-secondary);
font-style: italic;
}
.run-btn {
background: var(--bg-tertiary);
border: 1px solid var(--border-primary);
padding: 2px 6px;
border-radius: 2px;
color: var(--text-secondary);
cursor: pointer;
font-size: 0.75em;
font-family: inherit;
margin-left: 4px;
}
.run-btn:hover {
color: var(--text-primary);
background: var(--bg-primary);
}
.run-btn:disabled {
opacity: 0.6;
cursor: not-allowed;
}
.copy-btn {
background: var(--bg-tertiary);
border: 1px solid var(--border-primary);
padding: 2px 6px;
border-radius: 2px;
color: var(--text-secondary);
cursor: pointer;
font-size: 0.75em;
font-family: inherit;
margin-left: 4px;
}
.copy-btn:hover {
color: var(--text-primary);
background: var(--bg-primary);
}
.copy-btn:disabled {
opacity: 0.6;
cursor: not-allowed;
}
.copy-btn.copied {
color: #4caf50;
background: var(--bg-primary);
border-color: #4caf50;
transition: all 0.2s ease;
}
.raw-btn {
background: var(--bg-tertiary);
border: 1px solid var(--border-primary);
padding: 2px 6px;
border-radius: 2px;
color: var(--text-secondary);
cursor: pointer;
font-size: 0.75em;
font-family: inherit;
margin-left: 4px;
text-decoration: none;
display: inline-block;
}
.raw-btn:hover {
color: var(--text-primary);
background: var(--bg-primary);
text-decoration: none;
}
.github-btn {
background: var(--bg-tertiary);
border: 1px solid var(--border-primary);
padding: 2px 6px;
border-radius: 2px;
color: var(--text-secondary);
cursor: pointer;
font-size: 0.75em;
font-family: inherit;
margin-left: 4px;
text-decoration: none;
display: inline-block;
}
.github-btn:hover {
color: var(--text-primary);
background: var(--bg-primary);
text-decoration: none;
}
.hf-btn {
background: var(--bg-tertiary);
border: 1px solid var(--border-primary);
padding: 2px 6px;
border-radius: 2px;
color: var(--text-secondary);
cursor: pointer;
font-size: 0.75em;
font-family: inherit;
margin-left: 4px;
text-decoration: none;
display: inline-block;
}
.hf-btn:hover {
color: var(--text-primary);
background: var(--bg-primary);
text-decoration: none;
}
.output-stale {
opacity: 0.5;
position: relative;
}
.output-stale::after {
content: '⏳ updating...';
position: absolute;
top: 8px;
right: 8px;
background: var(--bg-secondary);
padding: 4px 8px;
border-radius: 2px;
font-size: 0.75em;
color: var(--text-secondary);
border: 1px solid var(--border-primary);
}
h1,
h2,
h3,
h4,
h5,
h6 {
margin-top: 1.5rem;
margin-bottom: 0.75rem;
color: var(--text-primary);
}
h1 {
margin-top: 0;
margin-bottom: 1rem;
}
p {
margin: 0.75rem 0;
color: var(--text-primary);
}
a {
color: var(--text-link);
}
img {
max-width: 100%;
height: auto;
border-radius: 1px;
box-shadow: none;
}
pre,
code {
font-family: 'Cascadia Mono', 'Cascadia Code', 'JetBrains Mono', 'SF Mono', Monaco, 'Consolas', monospace;
font-size: var(--code-font-size);
}
.code-wrap {
position: relative;
}
.code-line-highlight {
display: none;
position: absolute;
left: 0;
right: 0;
height: 1.5em;
background: rgba(255, 235, 170, 0.35);
pointer-events: none;
border-left: 3px solid #f4c542;
}
.line-number {
cursor: pointer;
text-decoration: none;
color: var(--text-secondary);
padding: 0 0.25rem;
}
.line-number.selected {
background: rgba(255, 235, 170, 0.4);
color: var(--text-primary);
}
/* Line numbers */
.highlight-with-lines {
display: flex;
}
.line-numbers {
background: var(--bg-tertiary);
padding: var(--code-pad-y) 0.5rem;
font-family: 'Cascadia Mono', 'Cascadia Code', 'JetBrains Mono', 'SF Mono', Monaco, 'Consolas', monospace;
font-size: var(--code-font-size);
line-height: var(--code-line-height);
color: var(--text-secondary);
user-select: none;
text-align: right;
border-right: 1px solid var(--border-primary);
}
.line-numbers .line-number {
display: block;
line-height: var(--code-line-height);
}
.highlight-with-lines .highlight {
flex: 1;
}
.highlight .hll {
background-color: transparent;
}
/* don't conflict with our highlight */
.highlight pre {
white-space: pre;
margin: 0;
padding: var(--code-pad-y) 0.75rem;
line-height: var(--code-line-height);
}
/* Collapsed code styling */
.cell-code.collapsed {
display: none;
}
.cell-code.expanded {
display: block;
}
{
% if config.collapse_code %
}
.cell-code {
display: none;
}
{
% else %
}
.cell-code {
display: block;
border-bottom: 1px solid var(--border-primary);
}
{
% endif %
}
{
{
pygments_css
}
}
/* Ensure our code metrics override Pygments defaults */
.highlight pre {
white-space: pre;
margin: 0;
padding: var(--code-pad-y) 0.75rem !important;
line-height: var(--code-line-height) !important;
font-size: var(--code-font-size) !important;
font-family: 'Cascadia Mono', 'Cascadia Code', 'JetBrains Mono', 'SF Mono', Monaco, 'Consolas', monospace !important;
border: none;
}
.line-numbers {
line-height: var(--code-line-height) !important;
}
.line-numbers .line-number {
line-height: var(--code-line-height) !important;
}
/* Custom CSS from frontmatter */
{
{
config.custom_css
}
}
{
# Override code font size from frontmatter (accept number as px) #
}
{
% if config.code_font_size is not none %
}
{
% if config.code_font_size is string %
}
:root {
--code-font-size: {
{
config.code_font_size
}
}
;
}
{
% else %
}
:root {
--code-font-size: {
{
config.code_font_size
}
}
px;
}
{
% endif %
}
{
% endif %
}
/* Cursor for tools */
body[data-tool="arrow"] .main-content {
cursor: url('data:image/svg+xml;utf8,<svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="%23e53935" stroke-width="2"><path d="M12 19l7-7 3 3-7 7-3-3z"/><path d="M18 13l-1.5-7.5L2 2l3.5 14.5L13 18l5-5z"/><path d="M2 2l7.586 7.586"/><circle cx="11" cy="11" r="2"/></svg>') 12 12, crosshair;
}
body[data-tool="pen"] .main-content {
cursor: url('data:image/svg+xml;utf8,<svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="%23e53935" stroke-width="2"><path d="M12 19l7-7 3 3-7 7-3-3z"/><path d="M18 13l-1.5-7.5L2 2l3.5 14.5L13 18l5-5z"/><circle cx="4" cy="20" r="2" fill="%23e53935"/></svg>') 4 20, pointer;
}
body[data-tool="eraser"] .main-content {
cursor: url('data:image/svg+xml;utf8,<svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="%23e53935" stroke-width="2"><path d="M20 20H7l-7-7 7-7h13v14z"/><path d="M13 13l7-7"/><path d="M13 13L9 9"/></svg>') 12 12, auto;
}
/* Color picker styles */
.tools-section-title {
font-weight: bold;
color: var(--text-secondary);
font-size: 0.65rem;
margin: 0.75rem 0 0.5rem 0;
text-transform: uppercase;
letter-spacing: 0.5px;
}
.color-row {
display: grid;
grid-template-columns: repeat(6, 1fr);
gap: 0.25rem;
margin-bottom: 0.5rem;
}
.color-swatch {
width: 18px;
height: 18px;
border: 2px solid var(--border-primary);
border-radius: 3px;
cursor: pointer;
transition: all 0.2s ease;
position: relative;
}
.color-swatch:hover {
transform: scale(1.1);
border-color: var(--text-secondary);
}
.color-swatch.selected {
border-color: var(--text-primary);
box-shadow: 0 0 0 2px var(--text-link);
}
.color-swatch.selected::after {
content: '✓';
position: absolute;
top: 50%;
left: 50%;
transform: translate(-50%, -50%);
color: white;
font-size: 10px;
font-weight: bold;
text-shadow: 1px 1px 1px black;
}
.color-input {
width: 24px;
height: 24px;
border: 2px solid var(--border-primary);
border-radius: 3px;
cursor: pointer;
background: none;
padding: 0;
grid-column: span 2;
justify-self: center;
}
.color-input:hover {
border-color: var(--text-secondary);
}
/* Thickness slider styles */
.thickness-row {
display: flex;
align-items: center;
gap: 0.5rem;
margin-top: 0.75rem;
}
.thickness-slider {
flex: 1;
-webkit-appearance: none;
appearance: none;
height: 4px;
background: var(--border-primary);
border-radius: 2px;
outline: none;
opacity: 0.7;
transition: opacity 0.2s;
}
.thickness-slider:hover {
opacity: 1;
}
.thickness-slider::-webkit-slider-thumb {
-webkit-appearance: none;
appearance: none;
width: 12px;
height: 12px;
background: var(--text-link);
border-radius: 50%;
cursor: pointer;
}
.thickness-slider::-moz-range-thumb {
width: 12px;
height: 12px;
background: var(--text-link);
border-radius: 50%;
cursor: pointer;
border: none;
}
.thickness-value {
font-size: 0.7rem;
color: var(--text-secondary);
min-width: 20px;
text-align: right;
}
.highlight {
background: none !important;
}
/* Loading animations */
.loading-spinner {
display: inline-block;
width: 16px;
height: 16px;
border: 2px solid var(--border-primary);
border-radius: 50%;
border-top-color: var(--text-link);
animation: spin 1s linear infinite;
margin-right: 8px;
vertical-align: middle;
}
@keyframes spin {
to {
transform: rotate(360deg);
}
}
.loading-skeleton {
display: inline-block;
background: var(--bg-tertiary);
background: linear-gradient(90deg,
var(--bg-tertiary) 25%,
var(--bg-secondary) 50%,
var(--bg-tertiary) 75%);
background-size: 200% 100%;
animation: loading-shimmer 2s ease-in-out infinite;
border-radius: 2px;
height: 1em;
width: 80px;
vertical-align: middle;
}
@keyframes loading-shimmer {
0% {
background-position: -200% 0;
}
100% {
background-position: 200% 0;
}
}
/* Loading state for cell output */
.cell-output:has(.loading-spinner) {
opacity: 0.7;
background: var(--bg-secondary);
/* border-left: 3px solid var(--text-link); */
}
</style>
<script>
// --- Drag utilities ---
function clamp(val, min, max) { return Math.max(min, Math.min(max, val)); }
function restorePosition(el, storageKey) {
try {
const raw = localStorage.getItem(storageKey);
if (!raw) return;
const pos = JSON.parse(raw);
if (typeof pos.left === 'number' && typeof pos.top === 'number') {
el.style.left = pos.left + 'px';
el.style.top = pos.top + 'px';
el.style.right = 'auto';
el.style.bottom = 'auto';
}
} catch (_) {}
}
function savePosition(el, storageKey) {
try {
const left = parseFloat(el.style.left || 'NaN');
const top = parseFloat(el.style.top || 'NaN');
if (!Number.isNaN(left) && !Number.isNaN(top)) {
localStorage.setItem(storageKey, JSON.stringify({ left, top }));
}
} catch (_) {}
}
function makeDraggable(el, storageKey, handleEl) {
let dragging = false;
let startX = 0, startY = 0; // cursor
let origLeft = 0, origTop = 0; // element
const onMove = (e) => {
if (!dragging) return;
const clientX = e.touches ? e.touches[0].clientX : e.clientX;
const clientY = e.touches ? e.touches[0].clientY : e.clientY;
const dx = clientX - startX;
const dy = clientY - startY;
const w = el.offsetWidth;
const h = el.offsetHeight;
const maxX = window.innerWidth - w;
const maxY = window.innerHeight - h;
const newLeft = clamp(origLeft + dx, 0, maxX);
const newTop = clamp(origTop + dy, 0, maxY);
el.style.left = newLeft + 'px';
el.style.top = newTop + 'px';
el.style.right = 'auto';
el.style.bottom = 'auto';
};
const endDrag = () => {
if (!dragging) return;
dragging = false;
document.removeEventListener('mousemove', onMove);
document.removeEventListener('mouseup', endDrag);
document.removeEventListener('touchmove', onMove);
document.removeEventListener('touchend', endDrag);
handleEl && (handleEl.style.cursor = 'grab');
savePosition(el, storageKey);
// ensure no-overlap constraint after a drag
try { layoutWidgetsStackedBottomRight(); } catch (_) {}
};
const startDrag = (e) => {
// Start from element's current on-screen rect
const elRect = el.getBoundingClientRect();
el.style.left = elRect.left + 'px';
el.style.top = elRect.top + 'px';
el.style.right = 'auto';
el.style.bottom = 'auto';
dragging = true;
startX = e.touches ? e.touches[0].clientX : e.clientX;
startY = e.touches ? e.touches[0].clientY : e.clientY;
origLeft = elRect.left;
origTop = elRect.top;
document.addEventListener('mousemove', onMove);
document.addEventListener('mouseup', endDrag);
document.addEventListener('touchmove', onMove, { passive: false });
document.addEventListener('touchend', endDrag);
handleEl && (handleEl.style.cursor = 'grabbing');
e.preventDefault();
};
(handleEl || el).addEventListener('mousedown', startDrag);
(handleEl || el).addEventListener('touchstart', startDrag, { passive: false });
// Apply any saved position on init
restorePosition(el, storageKey);
}
function toggleCell(cellId) {
const codeElement = document.getElementById('code-' + cellId);
const outputElement = document.getElementById('output-' + cellId);
if (codeElement) {
codeElement.classList.toggle('collapsed');
}
if (outputElement) {
outputElement.classList.toggle('collapsed');
}
updateIndicators(cellId);
encodeToolStateToUrl();
}
function toggleCode(cellId) {
const codeElement = document.getElementById('code-' + cellId);
if (codeElement) {
codeElement.classList.toggle('collapsed');
updateIndicators(cellId);
encodeToolStateToUrl();
}
}
function toggleOutput(cellId) {
const outputElement = document.getElementById('output-' + cellId);
if (outputElement) {
outputElement.classList.toggle('collapsed');
updateIndicators(cellId);
encodeToolStateToUrl();
}
}
function toggleUvLogs(headerElement) {
const contentElement = headerElement.nextElementSibling;
if (contentElement) {
const isCollapsed = contentElement.style.display === 'none';
contentElement.style.display = isCollapsed ? 'block' : 'none';
headerElement.textContent = isCollapsed ? '▼ UV Install Logs' : '▶ UV Install Logs';
// Update the header indicator if it exists
const uvLogsDiv = headerElement.parentElement;
if (uvLogsDiv && uvLogsDiv.id && uvLogsDiv.id.startsWith('uv-logs-')) {
const cellId = uvLogsDiv.id.replace('uv-logs-', '');
const indicatorElement = document.getElementById('uv-indicator-' + cellId);
if (indicatorElement) {
indicatorElement.textContent = isCollapsed ? '▼ uv-logs' : '▶ uv-logs';
}
}
}
}
function toggleUvLogsFromHeader(cellId) {
const uvLogsElement = document.getElementById('uv-logs-' + cellId);
const indicatorElement = document.getElementById('uv-indicator-' + cellId);
if (uvLogsElement) {
const headerElement = uvLogsElement.querySelector('.uv-logs-header');
const contentElement = uvLogsElement.querySelector('.uv-logs-content');
if (contentElement && headerElement) {
const isCollapsed = contentElement.style.display === 'none';
contentElement.style.display = isCollapsed ? 'block' : 'none';
headerElement.textContent = isCollapsed ? '▼ UV Install Logs' : '▶ UV Install Logs';
if (indicatorElement) {
indicatorElement.textContent = isCollapsed ? '▼ uv-logs' : '▶ uv-logs';
}
}
}
}
function updateIndicators(cellId) {
const codeElement = document.getElementById('code-' + cellId);
const outputElement = document.getElementById('output-' + cellId);
const indicators = document.querySelector(`[onclick*="${cellId}"]`)?.closest('.cell-header')?.querySelector('.collapse-indicators');
if (indicators) {
const codeCollapsed = codeElement && codeElement.classList.contains('collapsed');
const outputCollapsed = outputElement && outputElement.classList.contains('collapsed');
const codeIcon = codeCollapsed ? '▶' : '▼';
const outputIcon = outputCollapsed ? '▶' : '▼';
const codeSpan = indicators.querySelector('[onclick*="toggleCode"]');
const outputSpan = indicators.querySelector('[onclick*="toggleOutput"]');
if (codeSpan) codeSpan.innerHTML = `${codeIcon} code`;
if (outputSpan) outputSpan.innerHTML = `${outputIcon} output`;
}
}
function toggleTheme() {
const html = document.documentElement;
const currentTheme = html.getAttribute('data-theme');
const newTheme = currentTheme === 'dark' ? 'light' : 'dark';
html.setAttribute('data-theme', newTheme);
localStorage.setItem('uvnote-theme', newTheme);
updateThemeIcon();
updateUiDebug();
}
// Two panel code removed
function updateThemeIcon() {
const theme = document.documentElement.getAttribute('data-theme');
const toggle = document.querySelector('.theme-toggle');
if (toggle) {
toggle.textContent = theme === 'dark' ? 'light' : 'dark';
}
}
function setUiTheme(newUi) {
if (newUi !== 'default' && newUi !== 'none' && newUi !== 'monocolor') return;
const html = document.documentElement;
html.setAttribute('data-ui', newUi);
try { localStorage.setItem('uvnote-ui', newUi); } catch (_) {}
updateUiMenu();
updateUiDebug();
}
function updateUiMenu() {
const ui = document.documentElement.getAttribute('data-ui') || 'default';
const checks = {
default: document.getElementById('checkbox-ui-default'),
none: document.getElementById('checkbox-ui-none'),
monocolor: document.getElementById('checkbox-ui-monocolor')
};
if (checks.default) checks.default.textContent = ui === 'default' ? '☑' : '☐';
if (checks.none) checks.none.textContent = ui === 'none' ? '☑' : '☐';
if (checks.monocolor) checks.monocolor.textContent = ui === 'monocolor' ? '☑' : '☐';
}
function updateUiDebug() {
const ui = document.documentElement.getAttribute('data-ui') || 'default';
const color = document.documentElement.getAttribute('data-theme') || 'light';
const el = document.getElementById('ui-debug');
if (el) {
el.textContent = `UI: ${ui} | Color: ${color}`;
}
}
// Line selection and deep-linking
function clearLineSelections() {
try {
document.querySelectorAll('.code-line-highlight').forEach(el => { el.style.display = 'none'; });
document.querySelectorAll('.line-number.selected').forEach(el => el.classList.remove('selected'));
} catch (_) {}
}
let _selection = null; // { cellId, a, b }
function clearSelection(updateUrl) {
clearLineSelections();
_selection = null;
if (updateUrl) {
try {
const url = new URL(window.location.href);
url.searchParams.delete('cell');
url.searchParams.delete('line');
history.replaceState(null, '', url.toString());
} catch (_) {}
}
updateStateIndicator();
}
function selectCellLine(cellId, line, updateUrl) {
try {
// Ensure only one selection across the whole document
clearLineSelections();
const codeBox = document.getElementById(`code-${cellId}`);
if (!codeBox) return;
const pre = codeBox.querySelector('.highlight pre');
const overlay = document.getElementById(`line-highlight-${cellId}`);
const numbers = document.getElementById(`lines-${cellId}`);
if (!pre || !overlay) return;
// Measure line height directly from computed style
const preStyle = getComputedStyle(pre);
const padTop = parseFloat(preStyle.paddingTop || '0');
const lh = parseFloat(preStyle.lineHeight || '20');
// Position overlay
overlay.style.display = 'block';
overlay.style.height = `${lh}px`;
overlay.style.top = `${pre.offsetTop + padTop + (line - 1) * lh}px`;
// Update selected class in line numbers
if (numbers) {
numbers.querySelectorAll('.line-number').forEach(a => a.classList.remove('selected'));
const sel = numbers.querySelector(`.line-number[data-line="${line}"]`);
if (sel) sel.classList.add('selected');
}
if (updateUrl) {
const url = new URL(window.location.href);
url.searchParams.set('cell', cellId);
url.searchParams.set('line', String(line));
history.replaceState(null, '', url.toString());
}
_selection = { cellId, a: line, b: line };
updateStateIndicator();
} catch (e) { console.warn('selectCellLine error', e); }
}
function selectCellLines(cellId, startLine, endLine, updateUrl) {
try {
// normalize order
const a = Math.min(startLine, endLine);
const b = Math.max(startLine, endLine);
clearLineSelections();
const codeBox = document.getElementById(`code-${cellId}`);
if (!codeBox) return;
const pre = codeBox.querySelector('.highlight pre');
const overlay = document.getElementById(`line-highlight-${cellId}`);
const numbers = document.getElementById(`lines-${cellId}`);
if (!pre || !overlay) return;
const preStyle = getComputedStyle(pre);
const padTop = parseFloat(preStyle.paddingTop || '0');
const lh = parseFloat(preStyle.lineHeight || '20');
overlay.style.display = 'block';
overlay.style.top = `${pre.offsetTop + padTop + (a - 1) * lh}px`;
overlay.style.height = `${(b - a + 1) * lh}px`;
if (numbers) {
numbers.querySelectorAll('.line-number').forEach(a => a.classList.remove('selected'));
for (let i = a; i <= b; i++) {
const el = numbers.querySelector(`.line-number[data-line="${i}"]`);
if (el) el.classList.add('selected');
}
}
if (updateUrl) {
const url = new URL(window.location.href);
url.searchParams.set('cell', cellId);
if (a === b) url.searchParams.set('line', String(a));
else url.searchParams.set('line', `${a}-${b}`);
history.replaceState(null, '', url.toString());
}
_selection = { cellId, a, b };
updateStateIndicator();
} catch (e) { console.warn('selectCellLines error', e); }
}
// Drag-to-select support on line numbers
let _lineDrag = { active: false, cellId: null, start: 0 };
function onLineNumberMouseDown(e) {
const a = e.target.closest('.line-number');
if (!a) return;
e.preventDefault();
const cellId = a.dataset.cell;
const line = parseInt(a.dataset.line || '1', 10) || 1;
// Toggle off if this exact single line is already the only selection
const numbers = document.getElementById(`lines-${cellId}`);
if (numbers) {
const selected = Array.from(numbers.querySelectorAll('.line-number.selected')).map(n => parseInt(n.dataset.line||'0',10)).filter(Boolean);
if (selected.length === 1 && selected[0] === line) {
clearSelection(true);
return;
}
}
_lineDrag.active = true;
_lineDrag.cellId = cellId;
_lineDrag.start = line;
selectCellLines(_lineDrag.cellId, _lineDrag.start, _lineDrag.start, false);
}
function onDocMouseMove(e) {
if (!_lineDrag.active) return;
const el = document.elementFromPoint(e.clientX, e.clientY);
if (!el) return;
const a = el.closest && el.closest('.line-number');
if (!a) return;
if (a.dataset.cell !== _lineDrag.cellId) return;
const cur = parseInt(a.dataset.line || '1', 10) || 1;
selectCellLines(_lineDrag.cellId, _lineDrag.start, cur, false);
}
function onDocMouseUp(e) {
if (!_lineDrag.active) return;
const last = document.querySelector('.line-number.selected:last-of-type');
// finalize URL using the current selected range
const numbers = document.getElementById(`lines-${_lineDrag.cellId}`);
if (numbers) {
const selected = Array.from(numbers.querySelectorAll('.line-number.selected')).map(n => parseInt(n.dataset.line||'0',10)).filter(Boolean);
if (selected.length) {
const a = Math.min(...selected); const b = Math.max(...selected);
selectCellLines(_lineDrag.cellId, a, b, true);
}
}
_lineDrag.active = false; _lineDrag.cellId = null; _lineDrag.start = 0;
}
function applyLocationFromUrl() {
try {
const url = new URL(window.location.href);
const cell = url.searchParams.get('cell');
const lineParam = url.searchParams.get('line');
if (cell && lineParam) {
if (lineParam.includes('-')) {
const [a, b] = lineParam.split('-').map(x => parseInt(x, 10));
if (!Number.isNaN(a) && !Number.isNaN(b)) selectCellLines(cell, a, b, false);
} else {
const l = parseInt(lineParam, 10);
if (!Number.isNaN(l)) selectCellLine(cell, l, false);
}
}
// Apply tool parameters from URL
applyToolsFromUrl(url.searchParams);
// Cell states will be applied later in DOMContentLoaded with proper timing
const encodedCellStates = url.searchParams.get('cells');
console.log('Encoded cell states from URL:', encodedCellStates);
} catch (_) {}
}
function applyToolsFromUrl(params) {
try {
// Check if tools widget should be shown
const showTools = params.get('tools');
if (showTools === '1') {
// Mark that tool was loaded from URL
_urlLoadedTool = true;
// Apply color
const color = params.get('color');
if (color && /^[0-9a-fA-F]{6}$/.test(color)) {
setStoredArrowColor('#' + color);
}
// Apply thickness
const thickness = params.get('thickness');
if (thickness) {
const value = parseInt(thickness, 10);
if (value >= 1 && value <= 10) {
setStoredLineThickness(value);
}
}
// Don't override fadeout time for URL-loaded tools - let individual shapes decide
// Load shapes from URL
const encodedShapes = params.get('shapes');
if (encodedShapes) {
const decodedShapes = decodeShapesFromUrl(encodedShapes);
if (decodedShapes.length > 0) {
_shapes = decodedShapes;
saveShapes();
// Trigger render after overlay is initialized
setTimeout(() => {
renderOverlay();
}, 300);
}
}
// Wait for widgets to be initialized before showing tools
setTimeout(() => {
const toolsWidget = document.querySelector('.tools-widget');
const checkbox = document.getElementById('checkbox-tools');
if (toolsWidget && checkbox) {
toolsWidget.style.display = 'block';
checkbox.textContent = '☑';
localStorage.setItem('uvnote-widget-tools', 'visible');
}
// Apply active tool
const activeTool = params.get('tool');
if (activeTool && ['arrow', 'pen', 'eraser', 'spotlight'].includes(activeTool)) {
const toolBtn = Array.from(document.querySelectorAll('.tool-button')).find(btn => btn.textContent === activeTool);
if (toolBtn) {
toolBtn.click();
}
}
// Re-layout widgets after showing tools
layoutWidgetsStackedBottomRight();
}, 200);
}
} catch (_) {}
}
function captureInitialCellStates() {
const cells = document.querySelectorAll('.cell');
cells.forEach(cell => {
const cellId = cell.id.replace('cell-', '');
const codeEl = document.getElementById('code-' + cellId);
const outputEl = document.getElementById('output-' + cellId);
if (codeEl || outputEl) {
const state = {};
if (codeEl) {
state.c = codeEl.classList.contains('collapsed') ? 0 : 1;
}
if (outputEl) {
state.o = outputEl.classList.contains('collapsed') ? 0 : 1;
}
_initialCellStates[cellId] = state;
}
});
console.log('Captured initial cell states:', _initialCellStates);
}
function encodeCellStatesToUrl() {
// Get all cells and their collapse states
const cells = document.querySelectorAll('.cell');
const cellStates = {};
console.log('Found cells:', cells.length);
cells.forEach(cell => {
const cellId = cell.id.replace('cell-', '');
const codeEl = document.getElementById('code-' + cellId);
const outputEl = document.getElementById('output-' + cellId);
const initialState = _initialCellStates[cellId] || {};
console.log(`Encoding cell ${cellId}:`, {
codeEl: !!codeEl,
outputEl: !!outputEl,
codeCollapsed: codeEl ? codeEl.classList.contains('collapsed') : 'N/A',
outputCollapsed: outputEl ? outputEl.classList.contains('collapsed') : 'N/A',
initialState: initialState
});
if (codeEl || outputEl) {
const state = {};
let hasChanges = false;
if (codeEl) {
const currentCodeState = codeEl.classList.contains('collapsed') ? 0 : 1;
const initialCodeState = initialState.c;
// Only encode if different from initial state
if (initialCodeState !== undefined && currentCodeState !== initialCodeState) {
state.c = currentCodeState;
hasChanges = true;
}
}
if (outputEl) {
const currentOutputState = outputEl.classList.contains('collapsed') ? 0 : 1;
const initialOutputState = initialState.o;
// Only encode if different from initial state
if (initialOutputState !== undefined && currentOutputState !== initialOutputState) {
state.o = currentOutputState;
hasChanges = true;
}
}
// Only include cell if it has changes from initial state
if (hasChanges) {
cellStates[cellId] = state;
console.log(`Added cell ${cellId}:`, state);
}
}
});
console.log('Final cell states to encode:', cellStates);
// Return empty string if no changed cells
if (Object.keys(cellStates).length === 0) return '';
// Encode as compact base64 string
const encoded = btoa(JSON.stringify(cellStates));
console.log('Encoded cell states:', encoded);
return encoded;
}
function decodeCellStatesFromUrl(encodedStates) {
if (!encodedStates) return {};
try {
return JSON.parse(atob(encodedStates));
} catch (e) {
console.error('Failed to decode cell states:', e);
return {};
}
}
function applyCellStatesFromUrl(cellStates) {
console.log('Applying cell states from URL:', cellStates);
Object.entries(cellStates).forEach(([cellId, state]) => {
const codeEl = document.getElementById('code-' + cellId);
const outputEl = document.getElementById('output-' + cellId);
console.log(`Cell ${cellId}:`, {
codeEl: !!codeEl,
outputEl: !!outputEl,
state: state
});
if (codeEl && state.c !== undefined) {
if (state.c === 0) {
codeEl.classList.add('collapsed');
console.log(`Collapsed code for cell ${cellId}`, {
hasCollapsedClass: codeEl.classList.contains('collapsed'),
computedDisplay: getComputedStyle(codeEl).display,
classList: Array.from(codeEl.classList),
elementId: codeEl.id
});
} else {
codeEl.classList.remove('collapsed');
codeEl.classList.add('expanded'); // Explicitly add expanded class
console.log(`Expanded code for cell ${cellId}`, {
hasCollapsedClass: codeEl.classList.contains('collapsed'),
hasExpandedClass: codeEl.classList.contains('expanded'),
computedDisplay: getComputedStyle(codeEl).display,
classList: Array.from(codeEl.classList),
elementId: codeEl.id
});
}
}
if (outputEl && state.o !== undefined) {
if (state.o === 0) {
outputEl.classList.add('collapsed');
console.log(`Collapsed output for cell ${cellId}`);
} else {
outputEl.classList.remove('collapsed');
console.log(`Expanded output for cell ${cellId}`);
}
}
// Update visual indicators and force style recalculation
try {
updateIndicators(cellId);
// Force browser to recalculate styles
if (codeEl) {
codeEl.offsetHeight; // Force reflow
console.log(`After indicators update - code visible: ${getComputedStyle(codeEl).display !== 'none'}`);
}
if (outputEl) {
outputEl.offsetHeight; // Force reflow
console.log(`After indicators update - output visible: ${getComputedStyle(outputEl).display !== 'none'}`);
}
} catch (e) {
console.error(`Error updating indicators for cell ${cellId}:`, e);
}
});
}
function encodeShapesToUrl() {
// Encode shapes as compact base64 string
if (_shapes.length === 0) return '';
const shapeData = _shapes.map(shape => {
const baseData = {
ct: shape.createdAt, // creation timestamp
fo: shape.fadeoutTime || getFadeoutTime() // fadeout time for this shape
};
if (shape.type === 'arrow') {
return {
...baseData,
t: 'a',
x1: Math.round(shape.x1),
y1: Math.round(shape.y1),
x2: Math.round(shape.x2),
y2: Math.round(shape.y2),
c: shape.color.substring(1), // remove #
w: shape.width
};
} else if (shape.type === 'pen') {
return {
...baseData,
t: 'p',
pts: shape.points.map(p => [Math.round(p.x), Math.round(p.y)]),
c: shape.color.substring(1),
w: shape.width
};
} else if (shape.type === 'spotlight') {
return {
...baseData,
t: 's',
x: Math.round(shape.x),
y: Math.round(shape.y),
r: Math.round(shape.radius)
};
}
}).filter(Boolean);
return btoa(JSON.stringify(shapeData));
}
function decodeShapesFromUrl(encodedShapes) {
if (!encodedShapes) return [];
try {
const shapeData = JSON.parse(atob(encodedShapes));
return shapeData.map(data => {
const base = {
createdAt: data.ct || Date.now(), // use encoded timestamp or current time
fadeoutTime: data.fo || 0, // use encoded fadeout time or 0 (never fade)
opacity: 1.0
};
if (data.t === 'a') {
return {
...base,
type: 'arrow',
x1: data.x1,
y1: data.y1,
x2: data.x2,
y2: data.y2,
color: '#' + data.c,
width: data.w
};
} else if (data.t === 'p') {
return {
...base,
type: 'pen',
points: data.pts.map(([x, y]) => ({ x, y })),
color: '#' + data.c,
width: data.w
};
} else if (data.t === 's') {
return {
...base,
type: 'spotlight',
x: data.x,
y: data.y,
radius: data.r,
color: '#000000'
};
}
}).filter(Boolean);
} catch (e) {
console.error('Failed to decode shapes:', e);
return [];
}
}
function encodeToolStateToUrl() {
// Don't update URL during initialization
if (_isInitializing) {
return window.location.href;
}
const params = new URLSearchParams(window.location.search);
// Check if tools widget is visible and has an active tool
const toolsWidget = document.querySelector('.tools-widget');
const activeTool = document.body.dataset.tool;
const hasActiveTool = activeTool && activeTool !== 'none';
const toolsWidgetVisible = toolsWidget && getComputedStyle(toolsWidget).display !== 'none';
// Always handle shapes regardless of tool state
const encodedShapes = encodeShapesToUrl();
if (encodedShapes) {
params.set('shapes', encodedShapes);
} else {
params.delete('shapes');
}
// Always preserve existing cell states from URL if present
const existingCellStates = params.get('cells');
if (existingCellStates) {
// Keep existing cell states - don't re-encode from DOM
params.set('cells', existingCellStates);
} else {
// Only encode new cell states if none exist in URL
const encodedCellStates = encodeCellStatesToUrl();
if (encodedCellStates) {
params.set('cells', encodedCellStates);
}
}
if (toolsWidgetVisible && hasActiveTool) {
// Include tool params when widget is visible AND tool is active
params.set('tools', '1');
params.set('tool', activeTool);
// Get color (without # prefix)
const color = getArrowColor();
if (color && color.startsWith('#')) {
params.set('color', color.substring(1));
}
// Get thickness
const thickness = getLineThickness();
params.set('thickness', thickness.toString());
} else {
// Remove tool state params but keep shapes
params.delete('tools');
params.delete('tool');
params.delete('color');
params.delete('thickness');
params.delete('fadeout');
}
// Update URL without reloading
const newUrl = window.location.pathname + (params.toString() ? '?' + params.toString() : '') + window.location.hash;
window.history.replaceState(null, '', newUrl);
return window.location.href;
}
function resetLayout() {
try {
// Clear all uvnote-* keys
const allKeys = Object.keys(localStorage);
const uvnoteKeys = allKeys.filter(key => key.startsWith('uvnote-'));
uvnoteKeys.forEach(k => localStorage.removeItem(k));
} catch (_) {}
// Clear any active selection and remove URL params
try { clearSelection(true); } catch(_) {}
// Reset active tool if any
try { window.setActiveTool('none'); } catch(_) {}
// Clear shapes
try { _shapes = []; saveShapes(); } catch(_) {}
// Reset URL-loaded tool flag
try { _urlLoadedTool = false; } catch(_) {}
// Reset all cells to expanded state
try {
const cells = document.querySelectorAll('.cell');
cells.forEach(cell => {
const cellId = cell.id.replace('cell-', '');
const codeEl = document.getElementById('code-' + cellId);
const outputEl = document.getElementById('output-' + cellId);
if (codeEl) codeEl.classList.remove('collapsed');
if (outputEl) outputEl.classList.remove('collapsed');
updateIndicators(cellId);
});
} catch(_) {}
// Clear ALL URL parameters and reload with clean URL
try {
const cleanUrl = window.location.pathname + window.location.hash;
window.location.href = cleanUrl; // Use window.location.href instead of history.replaceState + reload
} catch (_) {
// Fallback - reload current page
location.reload();
}
}
function toggleMenu() {
const menuButton = document.querySelector('.menu-button');
if (menuButton) {
menuButton.classList.toggle('active');
}
}
function toggleWidget(widgetName) {
let widget;
let checkbox;
// Close the menu first
const menuButton = document.querySelector('.menu-button');
if (menuButton) {
menuButton.classList.remove('active');
}
switch(widgetName) {
case 'tools':
widget = document.querySelector('.tools-widget');
checkbox = document.getElementById('checkbox-tools');
break;
case 'file-explorer':
widget = document.querySelector('.file-explorer');
checkbox = document.getElementById('checkbox-file-explorer');
break;
case 'minimap':
widget = document.querySelector('.minimap');
checkbox = document.getElementById('checkbox-minimap');
break;
case 'status':
widget = document.querySelector('.status-widget');
checkbox = document.getElementById('checkbox-status');
break;
default:
return;
}
if (widget && checkbox) {
const isVisible = getComputedStyle(widget).display !== 'none';
widget.style.display = isVisible ? 'none' : 'block';
checkbox.textContent = isVisible ? '☐' : '☑';
// Save state to localStorage
try {
localStorage.setItem(`uvnote-widget-${widgetName}`, isVisible ? 'hidden' : 'visible');
} catch (_) {}
// Re-layout widgets after visibility change
try {
layoutWidgetsStackedBottomRight();
} catch (_) {}
// Update URL when tools widget visibility changes
if (widgetName === 'tools') {
encodeToolStateToUrl();
}
}
}
function initializeWidgetVisibility() {
const widgets = [
{ name: 'tools', selector: '.tools-widget' },
{ name: 'file-explorer', selector: '.file-explorer' },
{ name: 'minimap', selector: '.minimap' },
{ name: 'status', selector: '.status-widget' }
];
widgets.forEach(({ name, selector }) => {
const defaultState = name === 'status' ? 'visible' : 'hidden';
const savedState = localStorage.getItem(`uvnote-widget-${name}`) || defaultState;
const widget = document.querySelector(selector);
const checkbox = document.getElementById(`checkbox-${name}`);
if (widget && checkbox) {
const isVisible = savedState === 'visible';
widget.style.display = isVisible ? 'block' : 'none';
checkbox.textContent = isVisible ? '☑' : '☐';
}
});
}
// Close menu when clicking outside
document.addEventListener('click', function(event) {
const menuButton = document.querySelector('.menu-button');
// Don't close if clicking on a menu item (let the item handler close it)
if (menuButton && !menuButton.contains(event.target)) {
menuButton.classList.remove('active');
}
});
// Layout: stack widgets bottom-right and equalize widths
function hasCustomWidgetPositions() {
try {
return (
localStorage.getItem('uvnote-minimap-pos') ||
localStorage.getItem('uvnote-file-explorer-pos') ||
localStorage.getItem('uvnote-tools-pos')
);
} catch (_) { return false; }
}
function rectsOverlap(r1, r2) {
return !(r1.right <= r2.left || r2.right <= r1.left || r1.bottom <= r2.top || r2.bottom <= r1.top);
}
function widgetsOverlap(widgets) {
for (let i = 0; i < widgets.length; i++) {
const a = widgets[i];
const ra = a.getBoundingClientRect();
for (let j = i + 1; j < widgets.length; j++) {
const b = widgets[j];
const rb = b.getBoundingClientRect();
if (rectsOverlap(ra, rb)) return true;
}
}
return false;
}
function applyStackLayout(widgets, order) {
if (!widgets.length) return;
// Fixed equal width
const fixedWidth = 220;
widgets.forEach(el => { el.style.width = fixedWidth + 'px'; });
// Fit heights if needed to avoid overflow
const gap = 12;
const available = Math.max(0, window.innerHeight - 40 - gap * (order.length - 1));
const eachMax = Math.floor(available / order.length);
order.forEach(el => {
el.style.maxHeight = eachMax + 'px';
el.style.overflowY = 'auto';
});
// Stack bottom-up in the requested order
let bottomOffset = 20; // base gutter
order.forEach(el => {
el.style.left = 'auto';
el.style.top = 'auto';
el.style.right = '20px';
el.style.bottom = bottomOffset + 'px';
bottomOffset += el.offsetHeight + gap;
});
}
function layoutWidgetsStackedBottomRight() {
const minimap = document.querySelector('.minimap');
const fileExplorer = document.querySelector('.file-explorer');
const tools = document.querySelector('.tools-widget');
const status = document.querySelector('.status-widget');
const widgets = [minimap, fileExplorer, tools, status].filter(el => el && getComputedStyle(el).display !== 'none');
if (!widgets.length) return;
const order = [minimap, fileExplorer, tools, status].filter(Boolean).filter(el => getComputedStyle(el).display !== 'none');
// If user placed custom positions and there is no overlap, respect them.
if (hasCustomWidgetPositions() && !widgetsOverlap(widgets)) return;
applyStackLayout(widgets, order);
}
// Panel icon removed
let _minimapScrollContainer = null;
let _minimapScrollHandler = null;
function initMinimap() {
// Generate minimap content
const minimap = createMinimap();
document.body.appendChild(minimap);
// Make draggable (use title as handle)
const mTitle = minimap.querySelector('.minimap-title');
makeDraggable(minimap, 'uvnote-minimap-pos', mTitle);
// Attach scroll listener to window (two-panel removed)
_minimapScrollContainer = window;
if (_minimapScrollContainer) {
_minimapScrollHandler = () => updateMinimapActive();
if (_minimapScrollContainer === window) {
window.addEventListener('scroll', _minimapScrollHandler);
} else {
_minimapScrollContainer.addEventListener('scroll', _minimapScrollHandler);
}
}
updateMinimapActive();
}
function teardownMinimap() {
const minimap = document.querySelector('.minimap');
if (minimap && minimap.parentNode) minimap.parentNode.removeChild(minimap);
if (_minimapScrollContainer && _minimapScrollHandler) {
if (_minimapScrollContainer === window) {
window.removeEventListener('scroll', _minimapScrollHandler);
} else {
_minimapScrollContainer.removeEventListener('scroll', _minimapScrollHandler);
}
}
_minimapScrollContainer = null;
_minimapScrollHandler = null;
}
function initFileExplorer() {
// Generate file explorer content
const fileExplorer = createFileExplorer();
document.body.appendChild(fileExplorer);
}
function createMinimap() {
const minimap = document.createElement('div');
minimap.className = 'minimap';
const title = document.createElement('div');
title.className = 'minimap-title';
title.textContent = 'navigation';
minimap.appendChild(title);
// Find all headings and cells
const root = document.querySelector('.main-content') || document;
const headings = root.querySelectorAll('h1, h2, h3, h4, h5, h6');
const cells = root.querySelectorAll('.cell');
// Combine and sort by position
const items = [];
headings.forEach(heading => {
const id = heading.id || generateId(heading.textContent);
if (!heading.id) heading.id = id;
items.push({
element: heading,
type: 'heading',
level: parseInt(heading.tagName.charAt(1)),
text: heading.textContent.trim(),
id: id,
position: heading.getBoundingClientRect().top + window.scrollY
});
});
cells.forEach(cell => {
const header = cell.querySelector('.cell-header');
if (header) {
const id = cell.id || `cell-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`;
if (!cell.id) cell.id = id;
items.push({
element: cell,
type: 'cell',
text: header.textContent.trim(),
id: id,
position: cell.getBoundingClientRect().top + window.scrollY
});
}
});
// Sort by position
items.sort((a, b) => a.position - b.position);
// Create minimap items
items.forEach(item => {
const link = document.createElement('a');
link.className = `minimap-item ${item.type === 'heading' ? 'minimap-heading' : 'minimap-cell'}`;
if (item.type === 'heading') {
link.classList.add(`h${item.level}`);
}
link.textContent = item.text.length > 25 ? item.text.substring(0, 22) + '...' : item.text;
link.href = `#${item.id}`;
link.onclick = function(e) {
e.preventDefault();
item.element.scrollIntoView({ behavior: 'smooth', block: 'start' });
};
minimap.appendChild(link);
});
return minimap;
}
function generateId(text) {
return text.toLowerCase()
.replace(/[^a-z0-9]+/g, '-')
.replace(/^-+|-+$/g, '')
.substring(0, 20);
}
function updateMinimapActive() {
const minimapItems = document.querySelectorAll('.minimap-item');
const container = _minimapScrollContainer || window;
const containerRect = container === window ? null : container.getBoundingClientRect();
const scrollPos = (container === window ? window.scrollY : container.scrollTop) + 100; // Offset for better detection
let activeItem = null;
minimapItems.forEach(item => {
const targetId = item.getAttribute('href').substring(1);
const target = document.getElementById(targetId);
if (target) {
const rectTop = target.getBoundingClientRect().top;
const targetPos = (container === window)
? rectTop + window.scrollY
: rectTop - containerRect.top + container.scrollTop;
if (targetPos <= scrollPos) {
activeItem = item;
}
}
item.classList.remove('active');
});
if (activeItem) {
activeItem.classList.add('active');
}
}
function createFileExplorer() {
const fileExplorer = document.createElement('div');
fileExplorer.className = 'file-explorer';
const title = document.createElement('div');
title.className = 'file-explorer-title';
title.textContent = 'files';
fileExplorer.appendChild(title);
// Make draggable (use title as handle)
makeDraggable(fileExplorer, 'uvnote-file-explorer-pos', title);
// Scripts section
const scriptsSection = document.createElement('div');
scriptsSection.className = 'file-explorer-section';
const scriptsTitle = document.createElement('div');
scriptsTitle.className = 'file-explorer-section-title';
scriptsTitle.textContent = 'scripts';
scriptsSection.appendChild(scriptsTitle);
// Find all cells and list their script files (single panel)
const root = document.querySelector('.main-content') || document;
const cells = root.querySelectorAll('.cell');
cells.forEach(cell => {
const header = cell.querySelector('.cell-header');
if (header) {
const cellText = header.textContent.trim();
const cellMatch = cellText.match(/Cell: ([a-zA-Z_][a-zA-Z0-9_]*)/);
if (cellMatch) {
const cellId = cellMatch[1];
const scriptItem = document.createElement('div');
scriptItem.className = 'file-explorer-item script';
scriptItem.textContent = `${cellId}.py`;
scriptItem.onclick = function() {
cell.scrollIntoView({ behavior: 'smooth', block: 'start' });
};
scriptsSection.appendChild(scriptItem);
}
}
});
fileExplorer.appendChild(scriptsSection);
// Artifacts section
const artifactsSection = document.createElement('div');
artifactsSection.className = 'file-explorer-section';
const artifactsTitle = document.createElement('div');
artifactsTitle.className = 'file-explorer-section-title';
artifactsTitle.textContent = 'artifacts';
artifactsSection.appendChild(artifactsTitle);
// Find all artifact links (single panel)
const artifactsRoot = document.querySelector('.main-content') || document;
const artifacts = artifactsRoot.querySelectorAll('.artifact');
if (artifacts.length === 0) {
const noArtifacts = document.createElement('div');
noArtifacts.className = 'file-explorer-item artifact';
noArtifacts.textContent = '(none)';
noArtifacts.style.opacity = '0.5';
artifactsSection.appendChild(noArtifacts);
} else {
artifacts.forEach(artifact => {
const artifactItem = document.createElement('div');
artifactItem.className = 'file-explorer-item artifact';
artifactItem.textContent = artifact.textContent;
artifactItem.onclick = function() {
artifact.click();
};
artifactsSection.appendChild(artifactItem);
});
}
fileExplorer.appendChild(artifactsSection);
return fileExplorer;
}
function initStatusWidget() {
let el = document.querySelector('.status-widget');
if (!el) {
el = document.createElement('div');
el.className = 'status-widget';
el.id = 'status-widget';
el.textContent = 'ready — Esc';
document.body.appendChild(el);
}
}
// Tools widget
let _cursorX = 0;
let _cursorY = 0;
let _cursorVisible = false;
function setActiveTool(tool) {
if (!tool || tool === 'none') {
document.body.dataset.tool = 'none';
localStorage.setItem('uvnote-active-tool', 'none');
setOverlayActive(false);
_cursorVisible = false;
// Remove active class from all tool buttons when deactivating
const toolButtons = document.querySelectorAll('.tools-widget .tool-button');
toolButtons.forEach(btn => btn.classList.remove('active'));
updateStateIndicator();
encodeToolStateToUrl();
return;
}
document.body.dataset.tool = tool;
localStorage.setItem('uvnote-active-tool', tool);
setOverlayActive(true);
_cursorVisible = true;
updateStateIndicator();
encodeToolStateToUrl();
}
// Make setActiveTool globally accessible for ESC key handler
window.setActiveTool = setActiveTool;
function getArrowColor() {
const saved = localStorage.getItem('uvnote-arrow-color');
if (saved) return saved;
return '#e53935'; // Default red color
}
function setStoredArrowColor(color) {
try { localStorage.setItem('uvnote-arrow-color', color); } catch (_) {}
}
function getLineThickness() {
const saved = localStorage.getItem('uvnote-line-thickness');
if (saved) return parseInt(saved, 10);
return 6; // default thickness
}
function setStoredLineThickness(thickness) {
try { localStorage.setItem('uvnote-line-thickness', thickness); } catch (_) {}
}
function getFadeoutTime() {
const saved = localStorage.getItem('uvnote-fadeout-time');
if (saved) return parseInt(saved, 10);
return 5; // default 5 seconds
}
function setStoredFadeoutTime(seconds) {
try { localStorage.setItem('uvnote-fadeout-time', seconds); } catch (_) {}
}
function createToolsWidget() {
const tools = document.createElement('div');
tools.className = 'tools-widget';
const title = document.createElement('div');
title.className = 'tools-title';
title.textContent = 'tools';
tools.appendChild(title);
const row = document.createElement('div');
row.className = 'tools-row';
tools.appendChild(row);
// Arrow tool
const arrowBtn = document.createElement('div');
arrowBtn.className = 'tool-button';
arrowBtn.textContent = 'arrow';
arrowBtn.onclick = function() {
const isActive = arrowBtn.classList.contains('active');
if (isActive) {
arrowBtn.classList.remove('active');
setActiveTool('none');
} else {
tools.querySelectorAll('.tool-button').forEach(b => b.classList.remove('active'));
arrowBtn.classList.add('active');
setActiveTool('arrow');
}
};
row.appendChild(arrowBtn);
// Pen tool
const penBtn = document.createElement('div');
penBtn.className = 'tool-button';
penBtn.textContent = 'pen';
penBtn.onclick = function() {
const isActive = penBtn.classList.contains('active');
if (isActive) {
penBtn.classList.remove('active');
setActiveTool('none');
} else {
tools.querySelectorAll('.tool-button').forEach(b => b.classList.remove('active'));
penBtn.classList.add('active');
setActiveTool('pen');
}
};
row.appendChild(penBtn);
// Eraser tool
const eraseBtn = document.createElement('div');
eraseBtn.className = 'tool-button';
eraseBtn.textContent = 'eraser';
eraseBtn.onclick = function() {
const isActive = eraseBtn.classList.contains('active');
if (isActive) {
eraseBtn.classList.remove('active');
setActiveTool('none');
} else {
tools.querySelectorAll('.tool-button').forEach(b => b.classList.remove('active'));
eraseBtn.classList.add('active');
setActiveTool('eraser');
}
};
row.appendChild(eraseBtn);
// Spotlight tool
const spotlightBtn = document.createElement('div');
spotlightBtn.className = 'tool-button';
spotlightBtn.textContent = 'spotlight';
spotlightBtn.onclick = function() {
const isActive = spotlightBtn.classList.contains('active');
if (isActive) {
spotlightBtn.classList.remove('active');
setActiveTool('none');
} else {
tools.querySelectorAll('.tool-button').forEach(b => b.classList.remove('active'));
spotlightBtn.classList.add('active');
setActiveTool('spotlight');
}
};
row.appendChild(spotlightBtn);
// Clear all
const clearBtn = document.createElement('div');
clearBtn.className = 'tool-button';
clearBtn.textContent = 'clear';
clearBtn.onclick = function() {
_shapes = [];
saveShapes();
renderOverlay();
};
row.appendChild(clearBtn);
// We'll add the copy button at the end of the widget
// Restore active state from storage
const saved = localStorage.getItem('uvnote-active-tool') || 'none';
if (saved === 'arrow') {
arrowBtn.classList.add('active');
setActiveTool('arrow');
} else if (saved === 'pen') {
penBtn.classList.add('active');
setActiveTool('pen');
} else if (saved === 'eraser') {
eraseBtn.classList.add('active');
setActiveTool('eraser');
} else if (saved === 'spotlight') {
spotlightBtn.classList.add('active');
setActiveTool('spotlight');
}
// Color selector
const colorTitle = document.createElement('div');
colorTitle.className = 'tools-section-title';
colorTitle.textContent = 'color';
tools.appendChild(colorTitle);
const colorRow = document.createElement('div');
colorRow.className = 'tools-row color-row';
tools.appendChild(colorRow);
const swatchColors = [
// Primary colors
'#e53935', '#fb8c00', '#fdd835', '#43a047', '#1e88e5', '#8e24aa',
// Additional useful colors
'#ff5722', '#795548', '#607d8b', '#9c27b0',
// Grayscale
'#000000', '#424242', '#9e9e9e', '#ffffff'
];
const swatches = [];
swatchColors.forEach(c => {
const s = document.createElement('div');
s.className = 'color-swatch';
s.style.backgroundColor = c;
s.title = c;
s.onclick = () => {
setStoredArrowColor(c);
refreshColorUI(c);
if (_cursorVisible) renderOverlay();
encodeToolStateToUrl();
};
colorRow.appendChild(s);
swatches.push(s);
});
const colorInput = document.createElement('input');
colorInput.type = 'color';
colorInput.className = 'color-input';
colorInput.oninput = () => {
setStoredArrowColor(colorInput.value);
refreshColorUI(colorInput.value);
if (_cursorVisible) renderOverlay();
encodeToolStateToUrl();
};
colorRow.appendChild(colorInput);
function refreshColorUI(selected) {
const selectedHex = selected.startsWith('#') ? selected.toLowerCase() : rgbToHex(selected);
swatches.forEach((s, i) => {
const swatchHex = swatchColors[i].toLowerCase();
if (swatchHex === selectedHex) {
s.classList.add('selected');
} else {
s.classList.remove('selected');
}
});
try {
colorInput.value = selectedHex;
} catch (_) {}
}
function rgbToHex(rgb) {
const m = rgb.match(/rgba?\((\d+),\s*(\d+),\s*(\d+)\)/i);
if (!m) return '#000000';
const r = parseInt(m[1]).toString(16).padStart(2, '0');
const g = parseInt(m[2]).toString(16).padStart(2, '0');
const b = parseInt(m[3]).toString(16).padStart(2, '0');
return `#${r}${g}${b}`;
}
// Restore color selection
refreshColorUI(getArrowColor());
// Thickness slider
const thicknessTitle = document.createElement('div');
thicknessTitle.className = 'tools-section-title';
thicknessTitle.textContent = 'thickness';
tools.appendChild(thicknessTitle);
const thicknessRow = document.createElement('div');
thicknessRow.className = 'thickness-row';
tools.appendChild(thicknessRow);
const thicknessSlider = document.createElement('input');
thicknessSlider.type = 'range';
thicknessSlider.className = 'thickness-slider';
thicknessSlider.min = '1';
thicknessSlider.max = '10';
thicknessSlider.value = getLineThickness();
const thicknessValue = document.createElement('span');
thicknessValue.className = 'thickness-value';
thicknessValue.textContent = thicknessSlider.value + 'px';
thicknessSlider.oninput = function() {
const value = parseInt(thicknessSlider.value, 10);
setStoredLineThickness(value);
thicknessValue.textContent = value + 'px';
if (_cursorVisible) renderOverlay();
encodeToolStateToUrl();
};
thicknessRow.appendChild(thicknessSlider);
thicknessRow.appendChild(thicknessValue);
// Fadeout time slider
const fadeoutTitle = document.createElement('div');
fadeoutTitle.className = 'tools-section-title';
fadeoutTitle.textContent = 'fadeout time';
tools.appendChild(fadeoutTitle);
const fadeoutRow = document.createElement('div');
fadeoutRow.className = 'thickness-row';
tools.appendChild(fadeoutRow);
const fadeoutSlider = document.createElement('input');
fadeoutSlider.type = 'range';
fadeoutSlider.className = 'thickness-slider';
fadeoutSlider.min = '0';
fadeoutSlider.max = '30';
fadeoutSlider.value = getFadeoutTime();
const fadeoutValue = document.createElement('span');
fadeoutValue.className = 'thickness-value';
fadeoutValue.textContent = fadeoutSlider.value === '0' ? 'never' : fadeoutSlider.value + 's';
fadeoutSlider.oninput = function() {
const value = parseInt(fadeoutSlider.value, 10);
setStoredFadeoutTime(value);
fadeoutValue.textContent = value === 0 ? 'never' : value + 's';
encodeToolStateToUrl();
};
fadeoutRow.appendChild(fadeoutSlider);
fadeoutRow.appendChild(fadeoutValue);
// Draggable behavior
makeDraggable(tools, 'uvnote-tools-pos', title);
return tools;
}
function initTools() {
const widget = createToolsWidget();
document.body.appendChild(widget);
}
function teardownTools() {
const w = document.querySelector('.tools-widget');
if (w && w.parentNode) w.parentNode.removeChild(w);
}
// --- Canvas overlay for tools ---
let _overlay = null;
let _overlayCtx = null;
let _overlayContainer = null; // window
let _overlayMode = 'single';
let _overlayResizeHandler = null;
let _overlayScrollHandler = null;
let _drawing = null; // current in-progress arrow {x1,y1,x2,y2}
let _shapes = []; // committed shapes for current mode
let _fadeTimer = null; // timer for fade animation
let _urlLoadedTool = false; // track if tool was loaded from URL
let _isInitializing = true; // prevent URL updates during initialization
let _initialCellStates = {}; // track initial cell states from page load
function getOverlayStorageKey() { return 'uvnote-shapes'; }
function loadShapes() {
try {
const raw = localStorage.getItem(getOverlayStorageKey());
_shapes = raw ? JSON.parse(raw) : [];
} catch (_) { _shapes = []; }
}
function saveShapes() {
try {
localStorage.setItem(getOverlayStorageKey(), JSON.stringify(_shapes));
// Always update URL when shapes change
encodeToolStateToUrl();
} catch (_) {}
}
function updateShapesFade() {
const now = Date.now();
let needsUpdate = false;
for (let i = _shapes.length - 1; i >= 0; i--) {
const shape = _shapes[i];
if (!shape.createdAt) continue; // Skip old shapes without timestamps
// Use individual shape's fadeout time, or global if not set
const shapesFadeoutSeconds = shape.fadeoutTime !== undefined ? shape.fadeoutTime : getFadeoutTime();
// Skip fading if fadeout is disabled for this shape
if (shapesFadeoutSeconds === 0) continue;
const fadeStartTime = Math.max(0, (shapesFadeoutSeconds - 2) * 1000); // Start fading 2s before end
const fadeEndTime = shapesFadeoutSeconds * 1000; // Fully gone after specified time
const age = now - shape.createdAt;
if (age >= fadeEndTime) {
// Remove completely faded shapes
_shapes.splice(i, 1);
needsUpdate = true;
} else if (age >= fadeStartTime) {
// Update opacity for fading shapes
const fadeProgress = (age - fadeStartTime) / (fadeEndTime - fadeStartTime);
const newOpacity = 1 - fadeProgress;
if (Math.abs(shape.opacity - newOpacity) > 0.01) {
shape.opacity = newOpacity;
needsUpdate = true;
}
}
}
if (needsUpdate) {
saveShapes();
renderOverlay();
// Update URL to remove faded shapes
encodeToolStateToUrl();
}
}
function getContentContainer() { return window; }
function updateOverlayModeAndContainer() {
_overlayContainer = window;
_overlayMode = 'single';
}
function updateOverlayBounds() {
if (!_overlay) return;
if (_overlayContainer === window) {
_overlay.style.position = 'fixed';
_overlay.style.left = '0px';
_overlay.style.top = '0px';
_overlay.width = window.innerWidth;
_overlay.height = window.innerHeight;
} else {
const rect = _overlayContainer.getBoundingClientRect();
_overlay.style.position = 'fixed';
_overlay.style.left = rect.left + 'px';
_overlay.style.top = rect.top + 'px';
_overlay.width = Math.max(0, Math.floor(rect.width));
_overlay.height = Math.max(0, Math.floor(rect.height));
}
renderOverlay();
}
function containerScrollLeft() {
return (_overlayContainer === window) ? (window.scrollX || 0) : (_overlayContainer.scrollLeft || 0);
}
function containerScrollTop() {
return (_overlayContainer === window) ? (window.scrollY || 0) : (_overlayContainer.scrollTop || 0);
}
function toCanvasCoords(clientX, clientY) {
const rect = _overlay.getBoundingClientRect();
return { x: clientX - rect.left, y: clientY - rect.top };
}
function onPointerDown(e) {
const tool = document.body.dataset.tool;
if (tool === 'arrow') {
startDrawArrow(e);
} else if (tool === 'pen') {
startDrawPen(e);
} else if (tool === 'eraser') {
eraseAt(e);
} else if (tool === 'spotlight') {
startDrawSpotlight(e);
}
}
function onPointerMove(e) {
// Update cursor position
const pt = toCanvasCoords(e.touches ? e.touches[0].clientX : e.clientX, e.touches ? e.touches[0].clientY : e.clientY);
_cursorX = pt.x;
_cursorY = pt.y;
if (!_drawing) {
// Just update cursor position and re-render
if (_cursorVisible) {
renderOverlay();
}
return;
}
if (_drawing.type === 'pen') {
moveDrawPen(e);
} else if (_drawing.type === 'spotlight') {
moveDrawSpotlight(e);
} else {
moveDrawArrow(e);
}
}
function onPointerEnter(e) {
_cursorVisible = document.body.dataset.tool !== 'none';
if (_cursorVisible) {
renderOverlay();
}
}
function onPointerLeave(e) {
_cursorVisible = false;
renderOverlay();
}
function onPointerUp(e) {
if (!_drawing) return;
if (_drawing.type === 'pen') {
endDrawPen();
} else if (_drawing.type === 'spotlight') {
endDrawSpotlight();
} else {
endDrawArrow();
}
}
function startDrawArrow(e) {
if (document.body.dataset.tool !== 'arrow') return;
const pt = toCanvasCoords(e.touches ? e.touches[0].clientX : e.clientX, e.touches ? e.touches[0].clientY : e.clientY);
_drawing = {
x1: pt.x + containerScrollLeft(),
y1: pt.y + containerScrollTop(),
x2: pt.x + containerScrollLeft(),
y2: pt.y + containerScrollTop(),
color: getArrowColor(),
width: getLineThickness()
};
renderOverlay();
e.preventDefault();
}
function moveDrawArrow(e) {
if (!_drawing) return;
const pt = toCanvasCoords(e.touches ? e.touches[0].clientX : e.clientX, e.touches ? e.touches[0].clientY : e.clientY);
_drawing.x2 = pt.x + containerScrollLeft();
_drawing.y2 = pt.y + containerScrollTop();
renderOverlay();
e.preventDefault();
}
function endDrawArrow() {
if (!_drawing) return;
_shapes.push({
type: 'arrow',
..._drawing,
createdAt: Date.now(),
fadeoutTime: getFadeoutTime(),
opacity: 1.0
});
_drawing = null;
saveShapes();
renderOverlay();
}
function startDrawPen(e) {
if (document.body.dataset.tool !== 'pen') return;
const pt = toCanvasCoords(e.touches ? e.touches[0].clientX : e.clientX, e.touches ? e.touches[0].clientY : e.clientY);
_drawing = {
type: 'pen',
points: [{
x: pt.x + containerScrollLeft(),
y: pt.y + containerScrollTop()
}],
color: getArrowColor(),
width: getLineThickness()
};
renderOverlay();
e.preventDefault();
}
function moveDrawPen(e) {
if (!_drawing || _drawing.type !== 'pen') return;
const pt = toCanvasCoords(e.touches ? e.touches[0].clientX : e.clientX, e.touches ? e.touches[0].clientY : e.clientY);
_drawing.points.push({
x: pt.x + containerScrollLeft(),
y: pt.y + containerScrollTop()
});
renderOverlay();
e.preventDefault();
}
function endDrawPen() {
if (!_drawing || _drawing.type !== 'pen') return;
if (_drawing.points.length > 1) {
_shapes.push({
..._drawing,
createdAt: Date.now(),
fadeoutTime: getFadeoutTime(),
opacity: 1.0
});
}
_drawing = null;
saveShapes();
renderOverlay();
}
function startDrawSpotlight(e) {
if (document.body.dataset.tool !== 'spotlight') return;
const pt = toCanvasCoords(e.touches ? e.touches[0].clientX : e.clientX, e.touches ? e.touches[0].clientY : e.clientY);
_drawing = {
type: 'spotlight',
x: pt.x + containerScrollLeft(),
y: pt.y + containerScrollTop(),
radius: getLineThickness() * 20, // Use thickness to control spotlight size (bigger default)
color: getArrowColor()
};
renderOverlay();
e.preventDefault();
}
function moveDrawSpotlight(e) {
if (!_drawing || _drawing.type !== 'spotlight') return;
const pt = toCanvasCoords(e.touches ? e.touches[0].clientX : e.clientX, e.touches ? e.touches[0].clientY : e.clientY);
const dx = pt.x + containerScrollLeft() - _drawing.x;
const dy = pt.y + containerScrollTop() - _drawing.y;
_drawing.radius = Math.max(20, Math.sqrt(dx * dx + dy * dy)); // Minimum radius of 20
renderOverlay();
e.preventDefault();
}
function endDrawSpotlight() {
if (!_drawing || _drawing.type !== 'spotlight') return;
_shapes.push({
..._drawing,
createdAt: Date.now(),
fadeoutTime: getFadeoutTime(),
opacity: 1.0
});
_drawing = null;
saveShapes();
renderOverlay();
}
function distPointToSegment(px, py, x1, y1, x2, y2) {
const dx = x2 - x1, dy = y2 - y1;
if (dx === 0 && dy === 0) return Math.hypot(px - x1, py - y1);
const t = Math.max(0, Math.min(1, ((px - x1) * dx + (py - y1) * dy) / (dx*dx + dy*dy)));
const cx = x1 + t * dx, cy = y1 + t * dy;
return Math.hypot(px - cx, py - cy);
}
function eraseAt(e) {
const pt = toCanvasCoords(e.touches ? e.touches[0].clientX : e.clientX, e.touches ? e.touches[0].clientY : e.clientY);
const x = pt.x + containerScrollLeft();
const y = pt.y + containerScrollTop();
const threshold = 10; // pixels
for (let i = _shapes.length - 1; i >= 0; i--) {
const s = _shapes[i];
if (s.type === 'arrow') {
const d = distPointToSegment(x, y, s.x1, s.y1, s.x2, s.y2);
if (d <= threshold) {
_shapes.splice(i, 1);
saveShapes();
renderOverlay();
break;
}
} else if (s.type === 'pen' && s.points) {
// Check if click is near any line segment in the pen stroke
let minDist = Infinity;
for (let j = 1; j < s.points.length; j++) {
const d = distPointToSegment(x, y, s.points[j-1].x, s.points[j-1].y, s.points[j].x, s.points[j].y);
minDist = Math.min(minDist, d);
}
if (minDist <= threshold) {
_shapes.splice(i, 1);
saveShapes();
renderOverlay();
break;
}
}
}
e.preventDefault();
}
function drawArrow(ctx, x1, y1, x2, y2, color, width, opacity = 1.0) {
// Set opacity
const oldAlpha = ctx.globalAlpha;
ctx.globalAlpha = opacity;
ctx.strokeStyle = color;
ctx.fillStyle = color;
ctx.lineWidth = width;
ctx.lineCap = 'round';
ctx.lineJoin = 'round';
// Check if points are too close (initial state)
const dx = x2 - x1;
const dy = y2 - y1;
const distance = Math.sqrt(dx * dx + dy * dy);
if (distance < 5) {
// Draw just a small arrowhead pointing down-right when first clicked
const defaultAngle = Math.PI / 4; // 45 degrees (down-right)
const headLength = Math.min(15 + width * 1.5, 25);
const headAngle = Math.PI / 6;
// Calculate arrowhead points
const hx1 = x1 + headLength * Math.cos(defaultAngle - headAngle);
const hy1 = y1 + headLength * Math.sin(defaultAngle - headAngle);
const hx2 = x1 + headLength * Math.cos(defaultAngle + headAngle);
const hy2 = y1 + headLength * Math.sin(defaultAngle + headAngle);
// Draw arrowhead only
ctx.beginPath();
ctx.moveTo(x1, y1);
ctx.lineTo(hx1, hy1);
ctx.lineTo(hx2, hy2);
ctx.closePath();
ctx.fill();
} else {
// Normal arrow drawing - head at x1,y1, tail at x2,y2
const angle = Math.atan2(y1 - y2, x1 - x2);
const headLength = Math.min(15 + width * 1.5, 25);
const headAngle = Math.PI / 6;
// Calculate where the line should end (before the arrowhead)
const lineEndX = x1 - headLength * 0.8 * Math.cos(angle);
const lineEndY = y1 - headLength * 0.8 * Math.sin(angle);
// Draw the line from tail to near the head
ctx.beginPath();
ctx.moveTo(x2, y2);
ctx.lineTo(lineEndX, lineEndY);
ctx.stroke();
// Calculate arrowhead points
const hx1 = x1 - headLength * Math.cos(angle - headAngle);
const hy1 = y1 - headLength * Math.sin(angle - headAngle);
const hx2 = x1 - headLength * Math.cos(angle + headAngle);
const hy2 = y1 - headLength * Math.sin(angle + headAngle);
// Draw arrowhead
ctx.beginPath();
ctx.moveTo(x1, y1);
ctx.lineTo(hx1, hy1);
ctx.lineTo(hx2, hy2);
ctx.closePath();
ctx.fill();
}
// Restore opacity
ctx.globalAlpha = oldAlpha;
}
function drawPen(ctx, points, color, width, offX, offY, opacity = 1.0) {
if (!points || points.length < 2) return;
// Set opacity
const oldAlpha = ctx.globalAlpha;
ctx.globalAlpha = opacity;
ctx.strokeStyle = color;
ctx.lineWidth = width;
ctx.lineCap = 'round';
ctx.lineJoin = 'round';
ctx.beginPath();
ctx.moveTo(points[0].x - offX, points[0].y - offY);
for (let i = 1; i < points.length; i++) {
ctx.lineTo(points[i].x - offX, points[i].y - offY);
}
ctx.stroke();
// Restore opacity
ctx.globalAlpha = oldAlpha;
}
function drawAllSpotlights(ctx, spotlights, offX, offY) {
if (!spotlights || spotlights.length === 0) return;
ctx.save();
// Calculate the overall opacity based on all spotlights
const maxOpacity = Math.max(...spotlights.map(s => s.opacity || 1.0));
// Fill entire canvas with dark overlay
ctx.fillStyle = `rgba(0, 0, 0, ${0.7 * maxOpacity})`;
ctx.fillRect(0, 0, ctx.canvas.width, ctx.canvas.height);
// Cut out completely transparent holes for all spotlights
ctx.globalCompositeOperation = 'destination-out';
ctx.fillStyle = 'rgba(0, 0, 0, 1)'; // Solid black to ensure complete removal
for (const spotlight of spotlights) {
ctx.beginPath();
ctx.arc(spotlight.x - offX, spotlight.y - offY, spotlight.radius, 0, 2 * Math.PI);
ctx.fill();
}
ctx.restore();
}
function renderOverlay() {
if (!_overlay || !_overlayCtx) return;
_overlayCtx.clearRect(0, 0, _overlay.width, _overlay.height);
const offX = containerScrollLeft();
const offY = containerScrollTop();
// Draw non-spotlight shapes first
for (const s of _shapes) {
const opacity = s.opacity !== undefined ? s.opacity : 1.0;
if (s.type === 'arrow') {
drawArrow(_overlayCtx, s.x1 - offX, s.y1 - offY, s.x2 - offX, s.y2 - offY, s.color || '#f00', s.width || 2, opacity);
} else if (s.type === 'pen') {
drawPen(_overlayCtx, s.points, s.color || '#f00', s.width || 2, offX, offY, opacity);
}
}
// Draw current drawing (non-spotlight)
if (_drawing) {
if (_drawing.type === 'pen') {
drawPen(_overlayCtx, _drawing.points, _drawing.color, _drawing.width, offX, offY);
} else if (_drawing.type !== 'spotlight') {
drawArrow(_overlayCtx, _drawing.x1 - offX, _drawing.y1 - offY, _drawing.x2 - offX, _drawing.y2 - offY, _drawing.color, _drawing.width);
}
}
// Collect all spotlights (existing + current drawing + cursor preview)
const spotlights = [];
// Add existing spotlight shapes
for (const s of _shapes) {
if (s.type === 'spotlight') {
spotlights.push({
x: s.x,
y: s.y,
radius: s.radius,
opacity: s.opacity !== undefined ? s.opacity : 1.0
});
}
}
// Add current spotlight being drawn
if (_drawing && _drawing.type === 'spotlight') {
spotlights.push({
x: _drawing.x,
y: _drawing.y,
radius: _drawing.radius,
opacity: 1.0
});
}
// Add cursor preview spotlight if tool is active
if (_cursorVisible && !_drawing) {
const tool = document.body.dataset.tool;
if (tool === 'spotlight') {
const thickness = getLineThickness();
const radius = thickness * 20;
const cursorWorldX = _cursorX + containerScrollLeft();
const cursorWorldY = _cursorY + containerScrollTop();
spotlights.push({
x: cursorWorldX,
y: cursorWorldY,
radius: radius,
opacity: 0.8
});
}
}
// Draw all spotlights as a single overlay with multiple holes
drawAllSpotlights(_overlayCtx, spotlights, offX, offY);
// Draw cursor indicators for non-spotlight tools
if (_cursorVisible && !_drawing) {
const tool = document.body.dataset.tool;
const color = getArrowColor();
const thickness = getLineThickness();
if (tool !== 'spotlight') {
_overlayCtx.save();
_overlayCtx.fillStyle = color;
_overlayCtx.globalAlpha = 0.7;
if (tool === 'eraser') {
// Draw eraser indicator
_overlayCtx.strokeStyle = color;
_overlayCtx.lineWidth = 2;
_overlayCtx.beginPath();
_overlayCtx.arc(_cursorX, _cursorY, 10, 0, 2 * Math.PI);
_overlayCtx.stroke();
} else {
// Draw dot for pen/arrow
_overlayCtx.beginPath();
_overlayCtx.arc(_cursorX, _cursorY, thickness / 2, 0, 2 * Math.PI);
_overlayCtx.fill();
}
_overlayCtx.restore();
}
}
}
function setOverlayActive(active) {
if (!_overlay) initOverlay();
_overlay.style.pointerEvents = active ? 'auto' : 'none';
_overlay.style.cursor = active ? 'none' : 'auto';
// Re-render to ensure visibility aligns with content
renderOverlay();
}
function initOverlay() {
if (_overlay) return;
updateOverlayModeAndContainer();
_overlay = document.createElement('canvas');
_overlay.className = 'draw-overlay';
_overlayCtx = _overlay.getContext('2d');
document.body.appendChild(_overlay);
updateOverlayBounds();
loadShapes();
renderOverlay();
// Events
_overlay.addEventListener('mousedown', onPointerDown);
_overlay.addEventListener('mousemove', onPointerMove);
_overlay.addEventListener('mouseenter', onPointerEnter);
_overlay.addEventListener('mouseleave', onPointerLeave);
document.addEventListener('mouseup', onPointerUp);
_overlay.addEventListener('touchstart', onPointerDown, { passive: false });
_overlay.addEventListener('touchmove', onPointerMove, { passive: false });
document.addEventListener('touchend', onPointerUp);
_overlayResizeHandler = () => updateOverlayBounds();
window.addEventListener('resize', _overlayResizeHandler);
_overlayScrollHandler = () => renderOverlay();
window.addEventListener('scroll', _overlayScrollHandler);
// Start fade animation timer
_fadeTimer = setInterval(updateShapesFade, 100); // Update every 100ms for smooth fade
}
function rebindOverlayContainer() {
if (!_overlay) return;
// Remove old scroll handler
if (_overlayScrollHandler) { window.removeEventListener('scroll', _overlayScrollHandler); }
updateOverlayModeAndContainer();
updateOverlayBounds();
loadShapes();
renderOverlay();
_overlayScrollHandler = () => renderOverlay();
window.addEventListener('scroll', _overlayScrollHandler);
}
function teardownOverlay() {
if (!_overlay) return;
_overlay.removeEventListener('mousedown', onPointerDown);
_overlay.removeEventListener('mousemove', onPointerMove);
_overlay.removeEventListener('mouseenter', onPointerEnter);
_overlay.removeEventListener('mouseleave', onPointerLeave);
document.removeEventListener('mouseup', onPointerUp);
_overlay.removeEventListener('touchstart', onPointerDown);
_overlay.removeEventListener('touchmove', onPointerMove);
document.removeEventListener('touchend', onPointerUp);
if (_overlayResizeHandler) window.removeEventListener('resize', _overlayResizeHandler);
if (_overlayScrollHandler) {
if (_overlayContainer === window) {
window.removeEventListener('scroll', _overlayScrollHandler);
} else if (_overlayContainer) {
_overlayContainer.removeEventListener('scroll', _overlayScrollHandler);
}
}
if (_fadeTimer) {
clearInterval(_fadeTimer);
_fadeTimer = null;
}
if (_overlay.parentNode) _overlay.parentNode.removeChild(_overlay);
_overlay = null; _overlayCtx = null; _overlayContainer = null; _overlayResizeHandler = null; _overlayScrollHandler = null; _drawing = null;
}
function teardownFileExplorer() {
const fe = document.querySelector('.file-explorer');
if (fe && fe.parentNode) fe.parentNode.removeChild(fe);
}
function escapeHtml(text) {
const div = document.createElement('div');
div.textContent = text;
return div.innerHTML;
}
function runCell(cellId){
const btn=document.querySelector('.run-btn[onclick*="'+cellId+'"]');
const output=document.getElementById('output-'+cellId);
if(btn){btn.textContent='⏳ running...';btn.disabled=true;}
if(output){output.classList.add('output-stale');}
fetch('/run/'+cellId,{method:'POST'}).then(r=>r.json()).then(data=>{
if(output){
output.classList.remove('output-stale');
let html='';
if (data.stdout) {
html += '<div class="cell-stdout"><pre class="stdout-text">'
+ escapeHtml(data.stdout)
+ '</pre></div>';
}
console.log('UV Logs:', data);
if(data.stderr) {
// Split UV logs from regular stderr
const lines = data.stderr.split('\\n');
let uvLogs = [];
let regularLogs = [];
let inUvSection = true;
for (const line of lines) {
if (inUvSection) {
uvLogs.push(line);
if (line.startsWith('Installed ')) {
inUvSection = false;
}
} else {
regularLogs.push(line);
}
}
// If we never found "Installed", treat it all as regular stderr
if (inUvSection) {
html+='<div class="cell-stderr">'+escapeHtml(data.stderr)+'</div>';
} else {
const uvLogsStr = uvLogs.join('\\n');
const regularLogsStr = regularLogs.join('\\n').trim();
if (uvLogsStr) {
html+='<div class="uv-install-logs">';
html+='<div class="uv-logs-header" onclick="toggleUvLogs(this)">▶ UV Install Logs</div>';
html+='<div class="uv-logs-content" style="display: none;">'+escapeHtml(uvLogsStr)+'</div>';
html+='</div>';
}
if (regularLogsStr) {
html+='<div class="cell-stderr">'+escapeHtml(regularLogsStr)+'</div>';
}
}
}
output.innerHTML=html;
}
if(btn){btn.textContent='▶ run';btn.disabled=false;}
}).catch(e=>{
console.error('Run failed:',e);
if(output){output.classList.remove('output-stale');}
if(btn){btn.textContent='▶ run';btn.disabled=false;}
});
}
function copyCell(cellId){
// Try multiple selectors to find the code element
// Pygments generates .highlight > pre with spans, not wrapped in <code>
let codeElement = document.querySelector('#code-'+cellId+' .highlight pre');
if (!codeElement) {
codeElement = document.querySelector('#code-'+cellId+' pre');
}
if (!codeElement) {
codeElement = document.querySelector('#code-'+cellId+' code');
}
if (!codeElement) {
// Fallback to the code div itself
codeElement = document.getElementById('code-'+cellId);
}
const btn = document.querySelector('.copy-btn[onclick*="'+cellId+'"]');
if (!codeElement) {
console.error('Code element not found for cell:', cellId);
return;
}
if (!btn) {
console.error('Copy button not found for cell:', cellId);
return;
}
const codeText = codeElement.textContent;
if (navigator.clipboard && navigator.clipboard.writeText) {
navigator.clipboard.writeText(codeText).then(function() {
console.log('Clipboard copy successful');
btn.textContent = '✓ Copied!';
btn.classList.add('copied');
setTimeout(function() {
btn.textContent = 'Copy';
btn.classList.remove('copied');
}, 2000);
}).catch(function(err) {
console.warn('Clipboard copy failed:', err);
fallbackCopy();
});
} else {
console.log('Using fallback copy method');
fallbackCopy();
}
function fallbackCopy() {
const textarea = document.createElement('textarea');
textarea.value = codeText;
textarea.style.position = 'absolute';
textarea.style.left = '-9999px';
document.body.appendChild(textarea);
textarea.select();
try {
const success = document.execCommand('copy');
console.log('Fallback copy success:', success);
btn.textContent = '✓ Copied!';
btn.classList.add('copied');
setTimeout(function() {
btn.textContent = 'Copy';
btn.classList.remove('copied');
}, 2000);
} catch (err) {
console.error('Fallback copy failed:', err);
btn.textContent = 'Copy failed';
setTimeout(function() {
btn.textContent = 'Copy';
}, 2000);
}
document.body.removeChild(textarea);
}
}
// // Live reload functionality (robust SSE handling)
// (function(){
// if (!('EventSource' in window)) {
// console.warn('SSE not supported in this browser');
// return;
// }
// let source = new EventSource('/events');
// let isOpen = false;
// source.onopen = function(){ isOpen = true; console.log('SSE connected'); };
// source.onmessage = function(e){
// const msg=(e.data||'').trim(); if(!msg) return;
// console.log('SSE message:', msg);
// if (msg==='reload' || msg==='incremental') { location.reload(); }
// // Ignore 'loading' to avoid premature reload loops
// };
// source.onerror = function(e){
// // Let EventSource auto-reconnect instead of forcing a reload
// if (isOpen) console.warn('SSE error after open, retrying...', e);
// };
// window.addEventListener('beforeunload', function(){ try{source.close();}catch(_){} });
// })();
document.addEventListener('DOMContentLoaded', function() {
// Capture initial cell states before any modifications
captureInitialCellStates();
updateThemeIcon();
updateUiMenu();
updateUiDebug();
const widgetsEnabled = (document.documentElement.getAttribute('data-widgets') || 'on') === 'on';
if (widgetsEnabled) {
initMinimap();
initFileExplorer();
initTools();
initOverlay();
initStatusWidget();
initializeWidgetVisibility();
layoutWidgetsStackedBottomRight();
window.addEventListener('resize', layoutWidgetsStackedBottomRight);
}
// Apply deep-link selection if present
applyLocationFromUrl();
updateStateIndicator();
// Apply cell states from URL immediately
const url = new URL(window.location.href);
const encodedCellStates = url.searchParams.get('cells');
if (encodedCellStates) {
console.log('Applying cell states from URL...');
const cellStates = decodeCellStatesFromUrl(encodedCellStates);
// Use requestAnimationFrame to ensure DOM is ready
requestAnimationFrame(() => {
applyCellStatesFromUrl(cellStates);
// Clear initialization flag after cell states are applied
if (typeof _isInitializing !== 'undefined') {
_isInitializing = false;
}
});
} else {
// Clear initialization flag even if no cell states
if (typeof _isInitializing !== 'undefined') {
requestAnimationFrame(() => {
_isInitializing = false;
});
}
}
// Bind drag selection on line numbers
document.addEventListener('mousedown', onLineNumberMouseDown);
document.addEventListener('mousemove', onDocMouseMove);
document.addEventListener('mouseup', onDocMouseUp);
// Add ESC key handler to exit tools
document.addEventListener('keydown', function(e) {
if (e.key === 'Escape' || e.keyCode === 27) {
const currentTool = document.body.dataset.tool;
if (currentTool && currentTool !== 'none') {
// Deactivate the current tool
window.setActiveTool('none');
}
// Also clear any active line selection
clearSelection(true);
}
});
});
function updateStateIndicator() {
try {
const el = document.getElementById('status-widget');
if (!el) return;
const tool = document.body.dataset.tool || 'none';
if (tool && tool !== 'none') {
el.textContent = `tool: ${tool} — Esc`;
return;
}
if (_selection) {
const t = _selection.a === _selection.b ? `L${_selection.a}` : `L${_selection.a}-${_selection.b}`;
el.textContent = `selected: ${t} — Esc`;
return;
}
el.textContent = 'ready — Esc';
} catch (_) {}
}
</script>
</head>
<body>
<div class="controls">
<div class="controls-buttons">
<a href="index.html" class="back-button">← back</a>
<div class="theme-toggle" onclick="toggleTheme()">light</div>
<div class="reset-toggle" onclick="resetLayout()">reset</div>
<div class="menu-button" onclick="toggleMenu()">
menu ▼
<div class="menu-dropdown">
<div class="menu-item" onclick="setUiTheme('default')">
<span class="menu-checkbox" id="checkbox-ui-default">☑</span> Theme: default
</div>
<div class="menu-item" onclick="setUiTheme('none')">
<span class="menu-checkbox" id="checkbox-ui-none">☐</span> Theme: none
</div>
<div class="menu-item" onclick="setUiTheme('monocolor')">
<span class="menu-checkbox" id="checkbox-ui-monocolor">☐</span> Theme: monocolor
</div>
<div class="menu-item" onclick="toggleWidget('tools')">
<span class="menu-checkbox" id="checkbox-tools">☐</span> Tools
</div>
<div class="menu-item" onclick="toggleWidget('file-explorer')">
<span class="menu-checkbox" id="checkbox-file-explorer">☐</span> File Explorer
</div>
<div class="menu-item" onclick="toggleWidget('minimap')">
<span class="menu-checkbox" id="checkbox-minimap">☐</span> Table of Contents
</div>
<div class="menu-item" onclick="toggleWidget('status')">
<span class="menu-checkbox" id="checkbox-status">☑</span> Status Indicator
</div>
</div>
</div>
</div>
</div>
<div class="system-info">
<div class="system-info-header">Generated on:</div>
<div class="system-info-content">
Linux x86_64 | Linux-5.10.244-240.970.amzn2.x86_64-x86_64-with-glibc2.35
</div>
</div>
<div class="main-content">
<h1>GptOssExperts - OpenAI-style MoE</h1>
<h2>GPU Info</h2>
<div class="cell" id="cell-nv">
<div class="cell-header">
<span class="collapse-indicators">
<span onclick="toggleCode('nv')" style="cursor: pointer;">▼ code</span>
<span onclick="toggleOutput('nv')" style="cursor: pointer;">▼ output</span>
<span id="uv-indicator-nv" style="cursor: default; opacity: 0.3;">▶ uv-logs</span>
</span> |
Cell: nv | 0.24s
| <button class="run-btn" onclick="runCell('nv')">▶ run</button>
<button class="copy-btn" onclick="copyCell('nv')">Copy</button>
<a href="cells/nv.py" target="_blank" class="raw-btn">Raw</a>
<a href="https://github.com/huggingface/kernels-benchmarks/blob/main/openai_moe/impls/gpt_oss_moe.md" target="_blank" class="github-btn">GitHub</a>
<a href="https://huggingface.co/drbh/yamoe" target="_blank" class="hf-btn">🤗 HF</a>
</div>
<div id="code-nv" class="cell-code" data-lines="2">
<div class="code-wrap">
<div class="highlight"><pre><span></span><span class="kn">import</span><span class="w"> </span><span class="nn">subprocess</span>
<span class="nb">print</span><span class="p">(</span><span class="n">subprocess</span><span class="o">.</span><span class="n">run</span><span class="p">([</span><span class="s2">&quot;nvidia-smi&quot;</span><span class="p">],</span> <span class="n">capture_output</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> <span class="n">text</span><span class="o">=</span><span class="kc">True</span><span class="p">)</span><span class="o">.</span><span class="n">stdout</span><span class="p">)</span>
</pre></div>
<div class="code-line-highlight" id="line-highlight-nv"></div>
</div>
</div>
<div id="output-nv" class="cell-output">
<div class="cell-stdout"><pre class="stdout-text">Fri Oct 31 20:00:34 2025
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 570.195.03 Driver Version: 570.195.03 CUDA Version: 12.8 |
|-----------------------------------------+------------------------+----------------------+
| GPU Name Persistence-M | Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap | Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|=========================================+========================+======================|
| 0 NVIDIA L40S On | 00000000:4D:00.0 Off | 0 |
| N/A 34C P0 81W / 350W | 0MiB / 46068MiB | 18% Default |
| | | N/A |
+-----------------------------------------+------------------------+----------------------+
+-----------------------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=========================================================================================|
| No running processes found |
+-----------------------------------------------------------------------------------------+
</pre></div>
</div>
</div>
<h2>OpenAI-style MoE Benchmark (GptOssExperts Reference)</h2>
<div class="cell" id="cell-benchmark">
<div class="cell-header">
<span class="collapse-indicators">
<span onclick="toggleCode('benchmark')" style="cursor: pointer;">▼ code</span>
<span onclick="toggleOutput('benchmark')" style="cursor: pointer;">▼ output</span>
<span id="uv-indicator-benchmark" onclick="toggleUvLogsFromHeader('benchmark')" style="cursor: pointer;">▶ uv-logs</span>
</span> |
Cell: benchmark | 24.32s
| <button class="run-btn" onclick="runCell('benchmark')">▶ run</button>
<button class="copy-btn" onclick="copyCell('benchmark')">Copy</button>
<a href="cells/benchmark.py" target="_blank" class="raw-btn">Raw</a>
<a href="https://github.com/huggingface/kernels-benchmarks/blob/main/openai_moe/impls/gpt_oss_moe.md" target="_blank" class="github-btn">GitHub</a>
<a href="https://huggingface.co/drbh/yamoe" target="_blank" class="hf-btn">🤗 HF</a>
</div>
<div id="code-benchmark" class="cell-code" data-lines="81">
<div class="code-wrap">
<div class="highlight"><pre><span></span><span class="c1"># /// script</span>
<span class="c1"># requires-python = &quot;&gt;=3.10&quot;</span>
<span class="c1"># dependencies = [</span>
<span class="c1"># &quot;numpy&quot;,</span>
<span class="c1"># &quot;torch==2.8.0&quot;,</span>
<span class="c1"># &quot;kernels-benchmark-tools&quot;,</span>
<span class="c1"># &quot;kernels&quot;,</span>
<span class="c1"># ]</span>
<span class="c1">#</span>
<span class="c1"># [tool.uv.sources]</span>
<span class="c1"># kernels-benchmark-tools = { path = &quot;../../../../../tools&quot;, editable = true }</span>
<span class="c1"># kernels = { git = &quot;https://github.com/huggingface/kernels.git&quot; }</span>
<span class="c1"># ///</span>
<span class="kn">import</span><span class="w"> </span><span class="nn">torch</span>
<span class="kn">import</span><span class="w"> </span><span class="nn">sys</span>
<span class="kn">from</span><span class="w"> </span><span class="nn">kernels_benchmark_tools</span><span class="w"> </span><span class="kn">import</span> <span class="n">KernelTypeEnum</span><span class="p">,</span> <span class="n">run_benchmark</span>
<span class="kn">from</span><span class="w"> </span><span class="nn">kernels</span><span class="w"> </span><span class="kn">import</span> <span class="n">get_kernel</span>
<span class="c1"># Load yamoe to get GptOssExperts reference</span>
<span class="n">yamoe</span> <span class="o">=</span> <span class="n">get_kernel</span><span class="p">(</span><span class="s2">&quot;drbh/yamoe&quot;</span><span class="p">,</span> <span class="n">revision</span><span class="o">=</span><span class="s2">&quot;v0.2.0&quot;</span><span class="p">)</span>
<span class="n">GptOssExperts</span> <span class="o">=</span> <span class="n">yamoe</span><span class="o">.</span><span class="n">vendored</span><span class="o">.</span><span class="n">gpt_oss_mlp</span><span class="o">.</span><span class="n">GptOssExperts</span>
<span class="k">def</span><span class="w"> </span><span class="nf">gpt_oss_openai_moe</span><span class="p">(</span>
<span class="n">hidden_states</span><span class="p">,</span>
<span class="n">router_indices</span><span class="p">,</span>
<span class="n">routing_weights</span><span class="p">,</span>
<span class="n">gate_up_proj</span><span class="p">,</span>
<span class="n">gate_up_proj_bias</span><span class="p">,</span>
<span class="n">down_proj</span><span class="p">,</span>
<span class="n">down_proj_bias</span><span class="p">,</span>
<span class="p">):</span>
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> GptOssExperts reference implementation of OpenAI-style MoE.</span>
<span class="sd"> This is the reference model implementation from the original GPT OSS codebase.</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="n">B</span><span class="p">,</span> <span class="n">S</span><span class="p">,</span> <span class="n">H</span> <span class="o">=</span> <span class="n">hidden_states</span><span class="o">.</span><span class="n">shape</span>
<span class="n">E</span> <span class="o">=</span> <span class="n">routing_weights</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span>
<span class="c1"># Create a config object for GptOssExperts</span>
<span class="n">config</span> <span class="o">=</span> <span class="nb">type</span><span class="p">(</span><span class="s2">&quot;Config&quot;</span><span class="p">,</span> <span class="p">(),</span> <span class="p">{})()</span>
<span class="n">config</span><span class="o">.</span><span class="n">hidden_size</span> <span class="o">=</span> <span class="n">H</span>
<span class="n">config</span><span class="o">.</span><span class="n">intermediate_size</span> <span class="o">=</span> <span class="n">gate_up_proj</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span> <span class="o">//</span> <span class="mi">2</span> <span class="c1"># expert_dim / 2 = H</span>
<span class="n">config</span><span class="o">.</span><span class="n">num_local_experts</span> <span class="o">=</span> <span class="n">E</span>
<span class="c1"># Initialize model</span>
<span class="n">model</span> <span class="o">=</span> <span class="n">GptOssExperts</span><span class="p">(</span><span class="n">config</span><span class="p">)</span>
<span class="c1"># Set weights from benchmark inputs</span>
<span class="n">model</span><span class="o">.</span><span class="n">gate_up_proj</span><span class="o">.</span><span class="n">data</span> <span class="o">=</span> <span class="n">gate_up_proj</span>
<span class="n">model</span><span class="o">.</span><span class="n">gate_up_proj_bias</span><span class="o">.</span><span class="n">data</span> <span class="o">=</span> <span class="n">gate_up_proj_bias</span>
<span class="n">model</span><span class="o">.</span><span class="n">down_proj</span><span class="o">.</span><span class="n">data</span> <span class="o">=</span> <span class="n">down_proj</span>
<span class="n">model</span><span class="o">.</span><span class="n">down_proj_bias</span><span class="o">.</span><span class="n">data</span> <span class="o">=</span> <span class="n">down_proj_bias</span>
<span class="n">model</span> <span class="o">=</span> <span class="n">model</span><span class="o">.</span><span class="n">to</span><span class="p">(</span><span class="n">hidden_states</span><span class="o">.</span><span class="n">device</span><span class="p">)</span>
<span class="n">model</span><span class="o">.</span><span class="n">eval</span><span class="p">()</span>
<span class="c1"># Force GptOssExperts to use CPU path for correctness (matches naive_moe_ref behavior)</span>
<span class="c1"># The GPU path processes all experts which can lead to numerical differences</span>
<span class="c1"># CPU path explicitly uses router_indices like the reference implementation</span>
<span class="n">model</span><span class="o">.</span><span class="n">train</span><span class="p">()</span> <span class="c1"># Force CPU path</span>
<span class="c1"># Flatten routing_weights to [batch_seq, num_experts]</span>
<span class="n">routing_weights_flat</span> <span class="o">=</span> <span class="n">routing_weights</span><span class="o">.</span><span class="n">view</span><span class="p">(</span><span class="o">-</span><span class="mi">1</span><span class="p">,</span> <span class="n">E</span><span class="p">)</span>
<span class="c1"># Run forward pass</span>
<span class="k">with</span> <span class="n">torch</span><span class="o">.</span><span class="n">no_grad</span><span class="p">():</span>
<span class="n">output</span> <span class="o">=</span> <span class="n">model</span><span class="p">(</span><span class="n">hidden_states</span><span class="p">,</span> <span class="n">router_indices</span><span class="p">,</span> <span class="n">routing_weights_flat</span><span class="p">)</span>
<span class="n">model</span><span class="o">.</span><span class="n">eval</span><span class="p">()</span> <span class="c1"># Reset to eval mode</span>
<span class="k">return</span> <span class="n">output</span>
<span class="n">run_benchmark</span><span class="p">(</span>
<span class="n">kernel_type</span><span class="o">=</span><span class="n">KernelTypeEnum</span><span class="o">.</span><span class="n">OPENAI_MOE</span><span class="p">,</span>
<span class="n">impl_name</span><span class="o">=</span><span class="s2">&quot;gpt_oss_experts&quot;</span><span class="p">,</span>
<span class="n">impl_tags</span><span class="o">=</span><span class="p">{</span><span class="s2">&quot;family&quot;</span><span class="p">:</span> <span class="s2">&quot;reference&quot;</span><span class="p">,</span> <span class="s2">&quot;backend&quot;</span><span class="p">:</span> <span class="s2">&quot;pytorch&quot;</span><span class="p">},</span>
<span class="n">impl_func</span><span class="o">=</span><span class="n">gpt_oss_openai_moe</span><span class="p">,</span>
<span class="n">dtype</span><span class="o">=</span><span class="s2">&quot;float32&quot;</span><span class="p">,</span>
<span class="p">)</span>
</pre></div>
<div class="code-line-highlight" id="line-highlight-benchmark"></div>
</div>
</div>
<div id="output-benchmark" class="cell-output">
<div class="cell-stdout"><pre class="stdout-text">Running openai_moe benchmark on cuda with 8 workloads.
======================================================================
PROFILE TRACE: gpt_oss_experts | cuda_B1_S512_E2
======================================================================
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
Name Self CPU % Self CPU CPU total % CPU total CPU time avg Self CUDA Self CUDA % CUDA total CUDA time avg # of Calls
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
gpt_oss_experts 0.00% 0.000us 0.00% 0.000us 0.000us 10.211ms 197.81% 10.211ms 10.211ms 1
gpt_oss_experts 16.48% 2.023ms 99.94% 12.270ms 12.270ms 0.000us 0.00% 5.165ms 5.165ms 1
aten::matmul 0.22% 26.489us 3.82% 468.520us 39.043us 0.000us 0.00% 4.540ms 378.357us 12
aten::mm 2.36% 289.825us 3.60% 442.031us 36.836us 4.540ms 87.96% 4.540ms 378.357us 12
ampere_sgemm_128x64_nn 0.00% 0.000us 0.00% 0.000us 0.000us 3.078ms 59.62% 3.078ms 341.948us 9
void cutlass::Kernel2&lt;cutlass_80_simt_sgemm_128x64_8... 0.00% 0.000us 0.00% 0.000us 0.000us 1.457ms 28.23% 1.457ms 485.813us 3
aten::mul 1.42% 174.948us 2.34% 287.701us 11.988us 109.119us 2.11% 109.119us 4.547us 24
aten::add 1.61% 197.786us 3.85% 472.357us 26.242us 103.039us 2.00% 103.039us 5.724us 18
aten::index 1.73% 212.127us 2.86% 350.900us 29.242us 86.591us 1.68% 86.591us 7.216us 12
aten::index_add_ 0.51% 62.499us 0.79% 97.312us 16.219us 82.688us 1.60% 82.688us 13.781us 6
void at::native::indexFuncLargeIndex&lt;float, long, un... 0.00% 0.000us 0.00% 0.000us 0.000us 82.688us 1.60% 82.688us 13.781us 6
void at::native::elementwise_kernel&lt;128, 2, at::nati... 0.00% 0.000us 0.00% 0.000us 0.000us 80.511us 1.56% 80.511us 6.709us 12
aten::nonzero 2.20% 270.146us 6.58% 808.380us 89.820us 63.743us 1.23% 74.368us 8.263us 9
aten::clamp 0.98% 120.045us 1.63% 200.026us 16.669us 64.705us 1.25% 64.705us 5.392us 12
void at::native::elementwise_kernel&lt;128, 2, at::nati... 0.00% 0.000us 0.00% 0.000us 0.000us 64.705us 1.25% 64.705us 5.392us 12
aten::where 0.06% 7.400us 5.25% 644.007us 107.334us 0.000us 0.00% 60.384us 10.064us 6
aten::nonzero_numpy 0.11% 13.320us 5.19% 636.607us 106.101us 0.000us 0.00% 60.384us 10.064us 6
void at::native::vectorized_gather_kernel&lt;16, long&gt;(... 0.00% 0.000us 0.00% 0.000us 0.000us 60.063us 1.16% 60.063us 10.011us 6
void at::native::vectorized_elementwise_kernel&lt;4, at... 0.00% 0.000us 0.00% 0.000us 0.000us 56.800us 1.10% 56.800us 4.733us 12
Memcpy DtoH (Device -&gt; Pinned) 0.00% 0.000us 0.00% 0.000us 0.000us 50.911us 0.99% 50.911us 1.131us 45
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
Self CPU time total: 12.278ms
Self CUDA time total: 5.162ms
======================================================================
PROFILE TRACE: gpt_oss_experts | cuda_B1_S512_E4
======================================================================
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
Name Self CPU % Self CPU CPU total % CPU total CPU time avg Self CUDA Self CUDA % CUDA total CUDA time avg # of Calls
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
gpt_oss_experts 0.00% 0.000us 0.00% 0.000us 0.000us 13.933ms 229.38% 13.933ms 13.933ms 1
gpt_oss_experts 16.29% 2.560ms 99.97% 15.712ms 15.712ms 0.000us 0.00% 6.077ms 6.077ms 1
aten::matmul 0.30% 47.223us 5.17% 812.581us 33.858us 0.000us 0.00% 5.268ms 219.512us 24
aten::mm 3.09% 485.951us 4.87% 765.358us 31.890us 5.268ms 86.73% 5.268ms 219.512us 24
ampere_sgemm_128x64_nn 0.00% 0.000us 0.00% 0.000us 0.000us 5.213ms 85.81% 5.213ms 217.198us 24
aten::nonzero 2.45% 385.408us 7.89% 1.240ms 82.649us 112.163us 1.85% 134.498us 8.967us 15
aten::mul 2.03% 318.275us 3.36% 528.222us 11.005us 130.496us 2.15% 130.496us 2.719us 48
aten::add 2.25% 353.820us 3.74% 587.771us 16.327us 127.072us 2.09% 127.072us 3.530us 36
aten::where 0.08% 11.882us 7.49% 1.177ms 98.080us 0.000us 0.00% 120.705us 10.059us 12
aten::nonzero_numpy 0.15% 24.083us 7.41% 1.165ms 97.090us 0.000us 0.00% 120.705us 10.059us 12
aten::index 2.31% 363.442us 3.93% 617.030us 25.710us 110.145us 1.81% 110.145us 4.589us 24
void at::native::elementwise_kernel&lt;128, 2, at::nati... 0.00% 0.000us 0.00% 0.000us 0.000us 101.312us 1.67% 101.312us 4.221us 24
Memcpy DtoH (Device -&gt; Pinned) 0.00% 0.000us 0.00% 0.000us 0.000us 91.447us 1.51% 91.447us 1.051us 87
aten::clamp 1.32% 207.076us 2.26% 355.011us 14.792us 85.793us 1.41% 85.793us 3.575us 24
void at::native::elementwise_kernel&lt;128, 2, at::nati... 0.00% 0.000us 0.00% 0.000us 0.000us 85.793us 1.41% 85.793us 3.575us 24
aten::item 0.52% 81.620us 38.60% 6.066ms 84.255us 0.000us 0.00% 75.446us 1.048us 72
aten::_local_scalar_dense 2.00% 315.046us 38.08% 5.985ms 83.122us 75.446us 1.24% 75.446us 1.048us 72
aten::index_add_ 0.75% 118.511us 1.16% 182.084us 15.174us 72.926us 1.20% 72.926us 6.077us 12
void at::native::indexFuncLargeIndex&lt;float, long, un... 0.00% 0.000us 0.00% 0.000us 0.000us 72.926us 1.20% 72.926us 6.077us 12
void at::native::vectorized_gather_kernel&lt;16, long&gt;(... 0.00% 0.000us 0.00% 0.000us 0.000us 65.857us 1.08% 65.857us 5.488us 12
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
Self CPU time total: 15.717ms
Self CUDA time total: 6.074ms
======================================================================
PROFILE TRACE: gpt_oss_experts | cuda_B1_S1024_E2
======================================================================
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
Name Self CPU % Self CPU CPU total % CPU total CPU time avg Self CUDA Self CUDA % CUDA total CUDA time avg # of Calls
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
gpt_oss_experts 0.00% 0.000us 0.00% 0.000us 0.000us 12.540ms 148.48% 12.540ms 12.540ms 1
gpt_oss_experts 11.83% 1.734ms 99.96% 14.654ms 14.654ms 0.000us 0.00% 8.451ms 8.451ms 1
aten::matmul 0.16% 23.602us 3.00% 439.592us 36.633us 0.000us 0.00% 7.417ms 618.087us 12
aten::mm 1.78% 261.037us 2.84% 415.990us 34.666us 7.417ms 87.82% 7.417ms 618.087us 12
void cutlass::Kernel2&lt;cutlass_80_simt_sgemm_256x128_... 0.00% 0.000us 0.00% 0.000us 0.000us 4.532ms 53.65% 4.532ms 755.263us 6
ampere_sgemm_128x64_nn 0.00% 0.000us 0.00% 0.000us 0.000us 1.475ms 17.46% 1.475ms 491.509us 3
void cutlass::Kernel2&lt;cutlass_80_simt_sgemm_128x64_8... 0.00% 0.000us 0.00% 0.000us 0.000us 1.405ms 16.64% 1.405ms 468.490us 3
aten::mul 1.05% 153.262us 1.78% 261.173us 10.882us 197.791us 2.34% 197.791us 8.241us 24
aten::add 1.26% 184.574us 2.07% 304.007us 16.889us 188.543us 2.23% 188.543us 10.475us 18
aten::index_add_ 0.35% 50.951us 0.57% 83.553us 13.925us 169.408us 2.01% 169.408us 28.235us 6
void at::native::indexFuncLargeIndex&lt;float, long, un... 0.00% 0.000us 0.00% 0.000us 0.000us 169.408us 2.01% 169.408us 28.235us 6
void at::native::elementwise_kernel&lt;128, 2, at::nati... 0.00% 0.000us 0.00% 0.000us 0.000us 149.663us 1.77% 149.663us 12.472us 12
aten::index 1.27% 186.102us 2.16% 316.927us 26.411us 146.942us 1.74% 146.942us 12.245us 12
void at::native::vectorized_gather_kernel&lt;16, long&gt;(... 0.00% 0.000us 0.00% 0.000us 0.000us 117.440us 1.39% 117.440us 19.573us 6
aten::clamp 0.71% 104.743us 1.22% 178.924us 14.910us 110.912us 1.31% 110.912us 9.243us 12
void at::native::elementwise_kernel&lt;128, 2, at::nati... 0.00% 0.000us 0.00% 0.000us 0.000us 110.912us 1.31% 110.912us 9.243us 12
void at::native::vectorized_elementwise_kernel&lt;4, at... 0.00% 0.000us 0.00% 0.000us 0.000us 104.864us 1.24% 104.864us 8.739us 12
aten::nonzero 1.58% 232.211us 4.94% 724.348us 80.483us 69.633us 0.82% 81.377us 9.042us 9
aten::where 0.04% 6.259us 4.08% 597.684us 99.614us 0.000us 0.00% 66.816us 11.136us 6
aten::nonzero_numpy 0.08% 11.999us 4.03% 591.425us 98.571us 0.000us 0.00% 66.816us 11.136us 6
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
Self CPU time total: 14.659ms
Self CUDA time total: 8.446ms
======================================================================
PROFILE TRACE: gpt_oss_experts | cuda_B1_S1024_E4
======================================================================
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
Name Self CPU % Self CPU CPU total % CPU total CPU time avg Self CUDA Self CUDA % CUDA total CUDA time avg # of Calls
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
gpt_oss_experts 0.00% 0.000us 0.00% 0.000us 0.000us 18.317ms 174.31% 18.317ms 18.317ms 1
gpt_oss_experts 13.54% 2.761ms 99.97% 20.385ms 20.385ms 0.000us 0.00% 10.514ms 10.514ms 1
aten::matmul 0.23% 47.082us 4.02% 819.853us 34.161us 0.000us 0.00% 9.237ms 384.865us 24
aten::mm 2.37% 482.255us 3.79% 772.771us 32.199us 9.237ms 87.90% 9.237ms 384.865us 24
ampere_sgemm_128x64_nn 0.00% 0.000us 0.00% 0.000us 0.000us 6.282ms 59.78% 6.282ms 349.001us 18
void cutlass::Kernel2&lt;cutlass_80_simt_sgemm_128x64_8... 0.00% 0.000us 0.00% 0.000us 0.000us 2.944ms 28.01% 2.944ms 490.655us 6
aten::mul 1.50% 305.331us 2.55% 520.818us 10.850us 235.298us 2.24% 235.298us 4.902us 48
aten::add 1.72% 351.113us 2.86% 584.036us 16.223us 213.502us 2.03% 213.502us 5.931us 36
aten::index 1.95% 397.314us 3.28% 668.454us 27.852us 205.349us 1.95% 205.349us 8.556us 24
void at::native::elementwise_kernel&lt;128, 2, at::nati... 0.00% 0.000us 0.00% 0.000us 0.000us 166.720us 1.59% 166.720us 6.947us 24
aten::index_add_ 0.50% 101.340us 0.81% 165.573us 13.798us 155.585us 1.48% 155.585us 12.965us 12
void at::native::indexFuncLargeIndex&lt;float, long, un... 0.00% 0.000us 0.00% 0.000us 0.000us 155.585us 1.48% 155.585us 12.965us 12
void at::native::vectorized_gather_kernel&lt;16, long&gt;(... 0.00% 0.000us 0.00% 0.000us 0.000us 146.947us 1.40% 146.947us 12.246us 12
aten::nonzero 1.95% 398.176us 6.26% 1.276ms 85.090us 121.380us 1.16% 145.668us 9.711us 15
aten::clamp 1.04% 212.193us 1.79% 365.180us 15.216us 134.239us 1.28% 134.239us 5.593us 24
void at::native::elementwise_kernel&lt;128, 2, at::nati... 0.00% 0.000us 0.00% 0.000us 0.000us 134.239us 1.28% 134.239us 5.593us 24
aten::where 0.06% 11.340us 5.97% 1.216ms 101.373us 0.000us 0.00% 131.522us 10.960us 12
aten::nonzero_numpy 0.12% 24.140us 5.91% 1.205ms 100.428us 0.000us 0.00% 131.522us 10.960us 12
void at::native::vectorized_elementwise_kernel&lt;4, at... 0.00% 0.000us 0.00% 0.000us 0.000us 119.840us 1.14% 119.840us 4.993us 24
Memcpy DtoH (Device -&gt; Pinned) 0.00% 0.000us 0.00% 0.000us 0.000us 100.830us 0.96% 100.830us 1.159us 87
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
Self CPU time total: 20.390ms
Self CUDA time total: 10.509ms
======================================================================
PROFILE TRACE: gpt_oss_experts | cuda_B4_S512_E2
======================================================================
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
Name Self CPU % Self CPU CPU total % CPU total CPU time avg Self CUDA Self CUDA % CUDA total CUDA time avg # of Calls
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
gpt_oss_experts 0.00% 0.000us 0.00% 0.000us 0.000us 21.031ms 119.92% 21.031ms 21.031ms 1
gpt_oss_experts 7.59% 1.747ms 99.98% 23.024ms 23.024ms 0.000us 0.00% 17.548ms 17.548ms 1
aten::matmul 0.10% 23.660us 1.94% 446.020us 37.168us 0.000us 0.00% 14.659ms 1.222ms 12
aten::mm 1.17% 268.524us 1.83% 422.360us 35.197us 14.659ms 83.59% 14.659ms 1.222ms 12
void cutlass::Kernel2&lt;cutlass_80_simt_sgemm_256x128_... 0.00% 0.000us 0.00% 0.000us 0.000us 8.967ms 51.13% 8.967ms 1.495ms 6
ampere_sgemm_128x64_nn 0.00% 0.000us 0.00% 0.000us 0.000us 5.685ms 32.42% 5.685ms 947.562us 6
aten::add 0.82% 187.722us 1.36% 312.616us 17.368us 785.408us 4.48% 785.408us 43.634us 18
aten::mul 0.68% 156.369us 1.15% 264.222us 11.009us 674.688us 3.85% 674.688us 28.112us 24
void at::native::elementwise_kernel&lt;128, 2, at::nati... 0.00% 0.000us 0.00% 0.000us 0.000us 504.575us 2.88% 504.575us 42.048us 12
aten::index_add_ 0.22% 50.951us 0.37% 86.132us 14.355us 448.545us 2.56% 448.545us 74.757us 6
void at::native::indexFuncLargeIndex&lt;float, long, un... 0.00% 0.000us 0.00% 0.000us 0.000us 448.545us 2.56% 448.545us 74.757us 6
aten::clamp 0.46% 107.053us 0.80% 183.295us 15.275us 336.000us 1.92% 336.000us 28.000us 12
void at::native::elementwise_kernel&lt;128, 2, at::nati... 0.00% 0.000us 0.00% 0.000us 0.000us 336.000us 1.92% 336.000us 28.000us 12
void at::native::vectorized_elementwise_kernel&lt;4, at... 0.00% 0.000us 0.00% 0.000us 0.000us 314.239us 1.79% 314.239us 52.373us 6
void at::native::vectorized_elementwise_kernel&lt;4, at... 0.00% 0.000us 0.00% 0.000us 0.000us 280.833us 1.60% 280.833us 46.806us 6
aten::index 0.81% 185.806us 1.39% 320.548us 26.712us 259.102us 1.48% 259.102us 21.592us 12
void at::native::vectorized_elementwise_kernel&lt;4, at... 0.00% 0.000us 0.00% 0.000us 0.000us 258.944us 1.48% 258.944us 21.579us 12
void at::native::vectorized_gather_kernel&lt;16, long&gt;(... 0.00% 0.000us 0.00% 0.000us 0.000us 225.407us 1.29% 225.407us 37.568us 6
aten::sigmoid 0.16% 36.131us 0.27% 61.901us 10.317us 175.073us 1.00% 175.073us 29.179us 6
void at::native::vectorized_elementwise_kernel&lt;4, at... 0.00% 0.000us 0.00% 0.000us 0.000us 175.073us 1.00% 175.073us 29.179us 6
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
Self CPU time total: 23.030ms
Self CUDA time total: 17.537ms
======================================================================
PROFILE TRACE: gpt_oss_experts | cuda_B4_S512_E4
======================================================================
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
Name Self CPU % Self CPU CPU total % CPU total CPU time avg Self CUDA Self CUDA % CUDA total CUDA time avg # of Calls
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
gpt_oss_experts 0.00% 0.000us 0.00% 0.000us 0.000us 24.377ms 140.11% 24.377ms 24.377ms 1
gpt_oss_experts 10.50% 2.651ms 99.98% 25.237ms 25.237ms 0.000us 0.00% 17.408ms 17.408ms 1
aten::matmul 0.19% 47.519us 3.41% 860.801us 35.867us 0.000us 0.00% 15.185ms 632.705us 24
aten::mm 2.06% 521.061us 3.22% 813.282us 33.887us 15.185ms 87.28% 15.185ms 632.705us 24
void cutlass::Kernel2&lt;cutlass_80_simt_sgemm_256x128_... 0.00% 0.000us 0.00% 0.000us 0.000us 9.179ms 52.76% 9.179ms 764.922us 12
ampere_sgemm_128x64_nn 0.00% 0.000us 0.00% 0.000us 0.000us 3.124ms 17.96% 3.124ms 520.682us 6
void cutlass::Kernel2&lt;cutlass_80_simt_sgemm_128x64_8... 0.00% 0.000us 0.00% 0.000us 0.000us 2.871ms 16.50% 2.871ms 478.432us 6
aten::add 1.42% 359.495us 2.37% 598.003us 16.611us 427.713us 2.46% 427.713us 11.881us 36
aten::mul 1.23% 309.946us 2.09% 527.073us 10.981us 420.510us 2.42% 420.510us 8.761us 48
aten::index_add_ 0.40% 101.283us 0.66% 166.886us 13.907us 383.489us 2.20% 383.489us 31.957us 12
void at::native::indexFuncLargeIndex&lt;float, long, un... 0.00% 0.000us 0.00% 0.000us 0.000us 383.489us 2.20% 383.489us 31.957us 12
void at::native::elementwise_kernel&lt;128, 2, at::nati... 0.00% 0.000us 0.00% 0.000us 0.000us 343.712us 1.98% 343.712us 14.321us 24
aten::index 1.56% 393.991us 2.62% 662.158us 27.590us 337.086us 1.94% 337.086us 14.045us 24
void at::native::vectorized_gather_kernel&lt;16, long&gt;(... 0.00% 0.000us 0.00% 0.000us 0.000us 272.926us 1.57% 272.926us 22.744us 12
aten::clamp 0.84% 212.993us 1.44% 363.038us 15.127us 230.431us 1.32% 230.431us 9.601us 24
void at::native::elementwise_kernel&lt;128, 2, at::nati... 0.00% 0.000us 0.00% 0.000us 0.000us 230.431us 1.32% 230.431us 9.601us 24
void at::native::vectorized_elementwise_kernel&lt;4, at... 0.00% 0.000us 0.00% 0.000us 0.000us 223.071us 1.28% 223.071us 9.295us 24
aten::nonzero 1.57% 395.401us 5.00% 1.262ms 84.127us 128.836us 0.74% 156.164us 10.411us 15
aten::where 0.05% 12.011us 4.77% 1.205ms 100.378us 0.000us 0.00% 140.900us 11.742us 12
aten::nonzero_numpy 0.10% 25.021us 4.72% 1.193ms 99.377us 0.000us 0.00% 140.900us 11.742us 12
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
Self CPU time total: 25.242ms
Self CUDA time total: 17.398ms
======================================================================
PROFILE TRACE: gpt_oss_experts | cuda_B4_S1024_E2
======================================================================
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
Name Self CPU % Self CPU CPU total % CPU total CPU time avg Self CUDA Self CUDA % CUDA total CUDA time avg # of Calls
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
gpt_oss_experts 0.00% 0.000us 0.00% 0.000us 0.000us 40.556ms 109.47% 40.556ms 40.556ms 1
gpt_oss_experts 4.33% 1.794ms 99.85% 41.353ms 41.353ms 0.000us 0.00% 37.080ms 37.080ms 1
aten::matmul 0.06% 24.371us 1.08% 445.903us 37.159us 0.000us 0.00% 27.082ms 2.257ms 12
aten::mm 0.70% 291.738us 1.02% 421.532us 35.128us 27.082ms 73.10% 27.082ms 2.257ms 12
void cutlass::Kernel2&lt;cutlass_80_simt_sgemm_256x128_... 0.00% 0.000us 0.00% 0.000us 0.000us 27.079ms 73.09% 27.079ms 2.257ms 12
aten::mul 0.38% 159.199us 0.65% 268.178us 11.174us 2.983ms 8.05% 2.983ms 124.287us 24
aten::add 0.48% 198.424us 1.09% 451.763us 25.098us 2.404ms 6.49% 2.404ms 133.559us 18
aten::clamp 0.27% 112.290us 0.46% 189.433us 15.786us 2.392ms 6.46% 2.392ms 199.373us 12
void at::native::elementwise_kernel&lt;128, 2, at::nati... 0.00% 0.000us 0.00% 0.000us 0.000us 2.392ms 6.46% 2.392ms 199.373us 12
void at::native::vectorized_elementwise_kernel&lt;4, at... 0.00% 0.000us 0.00% 0.000us 0.000us 1.988ms 5.37% 1.988ms 165.669us 12
void at::native::elementwise_kernel&lt;128, 2, at::nati... 0.00% 0.000us 0.00% 0.000us 0.000us 1.629ms 4.40% 1.629ms 135.763us 12
aten::index_add_ 0.12% 50.103us 0.20% 84.453us 14.076us 899.456us 2.43% 899.456us 149.909us 6
void at::native::indexFuncLargeIndex&lt;float, long, un... 0.00% 0.000us 0.00% 0.000us 0.000us 899.456us 2.43% 899.456us 149.909us 6
void at::native::vectorized_elementwise_kernel&lt;4, at... 0.00% 0.000us 0.00% 0.000us 0.000us 774.912us 2.09% 774.912us 129.152us 6
void at::native::vectorized_elementwise_kernel&lt;4, at... 0.00% 0.000us 0.00% 0.000us 0.000us 733.217us 1.98% 733.217us 122.203us 6
aten::index 0.45% 187.302us 0.77% 318.787us 26.566us 712.767us 1.92% 712.767us 59.397us 12
void at::native::vectorized_gather_kernel&lt;16, long&gt;(... 0.00% 0.000us 0.00% 0.000us 0.000us 678.496us 1.83% 678.496us 113.083us 6
aten::sigmoid 0.09% 36.082us 0.15% 63.023us 10.504us 323.008us 0.87% 323.008us 53.835us 6
void at::native::vectorized_elementwise_kernel&lt;4, at... 0.00% 0.000us 0.00% 0.000us 0.000us 323.008us 0.87% 323.008us 53.835us 6
void at::native::elementwise_kernel&lt;128, 2, at::nati... 0.00% 0.000us 0.00% 0.000us 0.000us 261.631us 0.71% 261.631us 43.605us 6
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
Self CPU time total: 41.415ms
Self CUDA time total: 37.046ms
======================================================================
PROFILE TRACE: gpt_oss_experts | cuda_B4_S1024_E4
======================================================================
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
Name Self CPU % Self CPU CPU total % CPU total CPU time avg Self CUDA Self CUDA % CUDA total CUDA time avg # of Calls
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
gpt_oss_experts 0.00% 0.000us 0.00% 0.000us 0.000us 41.050ms 117.27% 41.050ms 41.050ms 1
gpt_oss_experts 6.46% 2.709ms 99.99% 41.912ms 41.912ms 0.000us 0.00% 35.025ms 35.025ms 1
aten::matmul 0.11% 47.590us 2.12% 888.873us 37.036us 0.000us 0.00% 29.051ms 1.210ms 24
aten::mm 1.28% 536.727us 2.01% 841.283us 35.053us 29.051ms 82.99% 29.051ms 1.210ms 24
void cutlass::Kernel2&lt;cutlass_80_simt_sgemm_256x128_... 0.00% 0.000us 0.00% 0.000us 0.000us 20.585ms 58.81% 20.585ms 1.372ms 15
ampere_sgemm_128x64_nn 0.00% 0.000us 0.00% 0.000us 0.000us 8.453ms 24.15% 8.453ms 939.204us 9
aten::add 0.88% 367.610us 1.45% 609.056us 16.918us 1.486ms 4.24% 1.486ms 41.264us 36
aten::mul 0.74% 309.128us 1.24% 518.283us 10.798us 1.380ms 3.94% 1.380ms 28.757us 48
void at::native::elementwise_kernel&lt;128, 2, at::nati... 0.00% 0.000us 0.00% 0.000us 0.000us 925.695us 2.64% 925.695us 38.571us 24
aten::index_add_ 0.24% 99.111us 0.40% 167.273us 13.939us 903.487us 2.58% 903.487us 75.291us 12
void at::native::indexFuncLargeIndex&lt;float, long, un... 0.00% 0.000us 0.00% 0.000us 0.000us 903.487us 2.58% 903.487us 75.291us 12
aten::clamp 0.51% 214.986us 0.87% 364.790us 15.200us 775.806us 2.22% 775.806us 32.325us 24
void at::native::elementwise_kernel&lt;128, 2, at::nati... 0.00% 0.000us 0.00% 0.000us 0.000us 775.806us 2.22% 775.806us 32.325us 24
aten::index 0.89% 373.269us 1.50% 629.207us 26.217us 670.881us 1.92% 670.881us 27.953us 24
void at::native::vectorized_elementwise_kernel&lt;4, at... 0.00% 0.000us 0.00% 0.000us 0.000us 631.200us 1.80% 631.200us 52.600us 12
void at::native::vectorized_gather_kernel&lt;16, long&gt;(... 0.00% 0.000us 0.00% 0.000us 0.000us 600.224us 1.71% 600.224us 50.019us 12
void at::native::vectorized_elementwise_kernel&lt;4, at... 0.00% 0.000us 0.00% 0.000us 0.000us 559.808us 1.60% 559.808us 46.651us 12
void at::native::vectorized_elementwise_kernel&lt;4, at... 0.00% 0.000us 0.00% 0.000us 0.000us 540.611us 1.54% 540.611us 22.525us 24
aten::sigmoid 0.17% 72.182us 0.29% 123.582us 10.298us 351.039us 1.00% 351.039us 29.253us 12
void at::native::vectorized_elementwise_kernel&lt;4, at... 0.00% 0.000us 0.00% 0.000us 0.000us 351.039us 1.00% 351.039us 29.253us 12
------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------ ------------
Self CPU time total: 41.917ms
Self CUDA time total: 35.005ms
impl wl p50(ms) ok
gpt_oss_experts cuda_B1_S1024_E2 3.79 True
gpt_oss_experts cuda_B1_S1024_E4 5.24 True
gpt_oss_experts cuda_B1_S512_E2 2.63 True
gpt_oss_experts cuda_B1_S512_E4 3.89 True
gpt_oss_experts cuda_B4_S1024_E2 13.28 True
gpt_oss_experts cuda_B4_S1024_E4 13.19 True
gpt_oss_experts cuda_B4_S512_E2 6.74 True
gpt_oss_experts cuda_B4_S512_E4 7.36 True
</pre></div>
<div class="uv-install-logs" id="uv-logs-benchmark">
<div class="uv-logs-header" onclick="toggleUvLogs(this)">▶ UV Install Logs</div>
<div class="uv-logs-content" style="display: none;">
Updating https://github.com/huggingface/kernels.git (HEAD)
Updated https://github.com/huggingface/kernels.git (10753bdcb99934587f52ff7fcb13cf5d23ab7b98)
Building kernels @ git+https://github.com/huggingface/kernels.git@10753bdcb99934587f52ff7fcb13cf5d23ab7b98
Built kernels @ git+https://github.com/huggingface/kernels.git@10753bdcb99934587f52ff7fcb13cf5d23ab7b98
Installed 52 packages in 205ms
</div>
</div>
<div class="cell-stderr">Fetching 6 files: 0%| | 0/6 [00:00&lt;?, ?it/s]
Fetching 6 files: 33%|███▎ | 2/6 [00:00&lt;00:00, 16.13it/s]
Fetching 6 files: 67%|██████▋ | 4/6 [00:00&lt;00:00, 7.33it/s]
Fetching 6 files: 100%|██████████| 6/6 [00:00&lt;00:00, 11.97it/s]</div>
<div class="cell-artifacts">
<h4>Artifacts:</h4>
<a href="artifacts/benchmark/openai_moe.jsonl" class="artifact" target="_blank">openai_moe.jsonl</a>
</div>
</div>
</div>
</div>
</body>
</html>