Skip to content

Commit ac18c9f

Browse files
committed
[js/webgpu] Donot record with computePassEncoder when capturing
1 parent 497b06f commit ac18c9f

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

js/web/lib/wasm/jsep/webgpu/program-manager.ts

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,6 @@ export class ProgramManager {
4141
): void {
4242
TRACE_FUNC_BEGIN(buildArtifact.programInfo.name);
4343
const device = this.backend.device;
44-
const computePassEncoder = this.backend.getComputePassEncoder();
4544
this.backend.writeTimestamp(this.backend.pendingDispatchNumber * 2);
4645
const entries = [];
4746
for (const input of inputs) {
@@ -68,11 +67,12 @@ export class ProgramManager {
6867
};
6968
const sessionCommandList = this.backend.capturedCommandList.get(this.backend.currentSessionId!);
7069
sessionCommandList!.push(commandInfo);
70+
} else {
71+
const computePassEncoder = this.backend.getComputePassEncoder();
72+
computePassEncoder.setPipeline(buildArtifact.computePipeline);
73+
computePassEncoder.setBindGroup(0, bindGroup);
74+
computePassEncoder.dispatchWorkgroups(...dispatchGroup);
7175
}
72-
73-
computePassEncoder.setPipeline(buildArtifact.computePipeline);
74-
computePassEncoder.setBindGroup(0, bindGroup);
75-
computePassEncoder.dispatchWorkgroups(...dispatchGroup);
7676
this.backend.writeTimestamp(this.backend.pendingDispatchNumber * 2 + 1);
7777
this.backend.pendingDispatchNumber++;
7878

0 commit comments

Comments
 (0)