dml gpu switch

This commit is contained in:
w-okada 2023-06-29 03:50:38 +09:00
parent 2bce23af19
commit 7b50b9f0f3
2 changed files with 105 additions and 70 deletions

File diff suppressed because one or more lines are too long

View File

@ -1,101 +1,136 @@
import React, { useMemo } from "react" import React, { useMemo } from "react";
import { useAppState } from "../../../001_provider/001_AppStateProvider" import { useAppState } from "../../../001_provider/001_AppStateProvider";
import { useAppRoot } from "../../../001_provider/001_AppRootProvider" import { useAppRoot } from "../../../001_provider/001_AppRootProvider";
export type ConvertProps = { export type ConvertProps = {
inputChunkNums: number[] inputChunkNums: number[];
} };
export const ConvertArea = (props: ConvertProps) => { export const ConvertArea = (props: ConvertProps) => {
const { setting, serverSetting, setWorkletNodeSetting, trancateBuffer } = useAppState() const { setting, serverSetting, setWorkletNodeSetting, trancateBuffer } = useAppState();
const { appGuiSettingState } = useAppRoot() const { appGuiSettingState } = useAppRoot();
const edition = appGuiSettingState.edition const edition = appGuiSettingState.edition;
const convertArea = useMemo(() => { const convertArea = useMemo(() => {
let nums: number[] let nums: number[];
if (!props.inputChunkNums) { if (!props.inputChunkNums) {
nums = [8, 16, 24, 32, 40, 48, 64, 80, 96, 112, 128, 192, 256, 320, 384, 448, 512, 576, 640, 704, 768, 832, 896, 960, 1024, 2048] nums = [8, 16, 24, 32, 40, 48, 64, 80, 96, 112, 128, 192, 256, 320, 384, 448, 512, 576, 640, 704, 768, 832, 896, 960, 1024, 2048];
} else { } else {
nums = props.inputChunkNums nums = props.inputChunkNums;
} }
const gpusEntry = [...serverSetting.serverSetting.gpus] const gpusEntry = [...serverSetting.serverSetting.gpus];
gpusEntry.push({ gpusEntry.push({
id: -1, id: -1,
name: "cpu", name: "cpu",
memory: 0 memory: 0,
}) });
const gpuSelect = edition.indexOf("onnxdirectML-cuda") >= 0 ? (
const onClassName = serverSetting.serverSetting.gpu == 0 ? "config-sub-area-button-active" : "config-sub-area-button";
const offClassName = serverSetting.serverSetting.gpu == 0 ? "config-sub-area-button" : "config-sub-area-button-active";
const gpuSelect =
edition.indexOf("onnxdirectML-cuda") >= 0 ? (
<div className="config-sub-area-control"> <div className="config-sub-area-control">
<div className="config-sub-area-control-title">GPU(dml):</div> <div className="config-sub-area-control-title">GPU(dml):</div>
<div className="config-sub-area-control-field"> <div className="config-sub-area-control-field">
<div className="config-sub-area-buttons left-padding-1"> <div className="config-sub-area-buttons">
<div onClick={ <div
async () => { onClick={async () => {
await serverSetting.updateServerSettings({ await serverSetting.updateServerSettings({
...serverSetting.serverSetting, gpu: serverSetting.serverSetting.gpu == 0 ? -1 : 0 ...serverSetting.serverSetting,
}) gpu: 0,
}} className="config-sub-area-button ">{serverSetting.serverSetting.gpu == 0 ? "on" : "off"}</div> });
}}
className={onClassName}
>
on
</div>
<div
onClick={async () => {
await serverSetting.updateServerSettings({
...serverSetting.serverSetting,
gpu: -1,
});
}}
className={offClassName}
>
off
</div>
</div> </div>
</div> </div>
</div> </div>
) : ( ) : (
<div className="config-sub-area-control"> <div className="config-sub-area-control">
<div className="config-sub-area-control-title">GPU:</div> <div className="config-sub-area-control-title">GPU:</div>
<div className="config-sub-area-control-field"> <div className="config-sub-area-control-field">
<select className="body-select" value={serverSetting.serverSetting.gpu} onChange={(e) => { <select
serverSetting.updateServerSettings({ ...serverSetting.serverSetting, gpu: Number(e.target.value) }) className="body-select"
}}> value={serverSetting.serverSetting.gpu}
{ onChange={(e) => {
gpusEntry.map(x => { serverSetting.updateServerSettings({ ...serverSetting.serverSetting, gpu: Number(e.target.value) });
return <option key={x.id} value={x.id}>{x.name}{x.name == "cpu" ? "" : `(${(x.memory / 1024 / 1024 / 1024).toFixed(0)}GB)`} </option> }}
}) >
} {gpusEntry.map((x) => {
return (
<option key={x.id} value={x.id}>
{x.name}
{x.name == "cpu" ? "" : `(${(x.memory / 1024 / 1024 / 1024).toFixed(0)}GB)`}{" "}
</option>
);
})}
</select> </select>
</div> </div>
</div> </div>
);
)
return ( return (
<div className="config-sub-area"> <div className="config-sub-area">
<div className="config-sub-area-control"> <div className="config-sub-area-control">
<div className="config-sub-area-control-title">CHUNK:</div> <div className="config-sub-area-control-title">CHUNK:</div>
<div className="config-sub-area-control-field"> <div className="config-sub-area-control-field">
<select className="body-select" value={setting.workletNodeSetting.inputChunkNum} onChange={(e) => { <select
setWorkletNodeSetting({ ...setting.workletNodeSetting, inputChunkNum: Number(e.target.value) }) className="body-select"
trancateBuffer() value={setting.workletNodeSetting.inputChunkNum}
serverSetting.updateServerSettings({ ...serverSetting.serverSetting, serverReadChunkSize: Number(e.target.value) }) onChange={(e) => {
}}> setWorkletNodeSetting({ ...setting.workletNodeSetting, inputChunkNum: Number(e.target.value) });
{ trancateBuffer();
nums.map(x => { serverSetting.updateServerSettings({ ...serverSetting.serverSetting, serverReadChunkSize: Number(e.target.value) });
return <option key={x} value={x}>{x} ({(x * 128 * 1000 / 48000).toFixed(1)} ms, {x * 128})</option> }}
}) >
} {nums.map((x) => {
return (
<option key={x} value={x}>
{x} ({((x * 128 * 1000) / 48000).toFixed(1)} ms, {x * 128})
</option>
);
})}
</select> </select>
</div> </div>
</div> </div>
<div className="config-sub-area-control"> <div className="config-sub-area-control">
<div className="config-sub-area-control-title">EXTRA:</div> <div className="config-sub-area-control-title">EXTRA:</div>
<div className="config-sub-area-control-field"> <div className="config-sub-area-control-field">
<select className="body-select" value={serverSetting.serverSetting.extraConvertSize} onChange={(e) => { <select
serverSetting.updateServerSettings({ ...serverSetting.serverSetting, extraConvertSize: Number(e.target.value) }) className="body-select"
trancateBuffer() value={serverSetting.serverSetting.extraConvertSize}
}}> onChange={(e) => {
{ serverSetting.updateServerSettings({ ...serverSetting.serverSetting, extraConvertSize: Number(e.target.value) });
[1024 * 4, 1024 * 8, 1024 * 16, 1024 * 32, 1024 * 64, 1024 * 128].map(x => { trancateBuffer();
return <option key={x} value={x}>{x}</option> }}
}) >
} {[1024 * 4, 1024 * 8, 1024 * 16, 1024 * 32, 1024 * 64, 1024 * 128].map((x) => {
return (
<option key={x} value={x}>
{x}
</option>
);
})}
</select> </select>
</div> </div>
</div> </div>
{gpuSelect} {gpuSelect}
</div> </div>
) );
}, [serverSetting.serverSetting, setting, serverSetting.updateServerSettings, setWorkletNodeSetting, edition]) }, [serverSetting.serverSetting, setting, serverSetting.updateServerSettings, setWorkletNodeSetting, edition]);
return convertArea;
return convertArea };
}