Compare commits
2 Commits
nvidia-sty
...
pre-wu-bla
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
3a8ff9490a | ||
|
|
a0b08f4c81 |
@@ -78,7 +78,9 @@ module VX_tensor_blackwell_core_block import VX_gpu_pkg::*; #(
|
||||
wire [`NW_WIDTH-1:0] writeback_wid;
|
||||
wire [4:0] writeback_rd;
|
||||
logic writeback_ready;
|
||||
wire [`NUM_THREADS-1:0][`XLEN-1:0] writeback_data;
|
||||
wire [7:0][`XLEN-1:0] tensor_writeback_data;
|
||||
wire [`NUM_THREADS-1:0][`XLEN-1:0] writeback_data =
|
||||
tensor_writeback_data[`NUM_THREADS-1:0];
|
||||
|
||||
wire metadata_valid = !metadata_queue_empty;
|
||||
wire bwgmma = metadata_valid &&
|
||||
@@ -135,14 +137,14 @@ module VX_tensor_blackwell_core_block import VX_gpu_pkg::*; #(
|
||||
.io_writeback_bits_last(writeback_last),
|
||||
.io_writeback_bits_wid(writeback_wid),
|
||||
.io_writeback_bits_rd(writeback_rd),
|
||||
.io_writeback_bits_data_0(writeback_data[0]),
|
||||
.io_writeback_bits_data_1(writeback_data[1]),
|
||||
.io_writeback_bits_data_2(writeback_data[2]),
|
||||
.io_writeback_bits_data_3(writeback_data[3]),
|
||||
.io_writeback_bits_data_4(writeback_data[4]),
|
||||
.io_writeback_bits_data_5(writeback_data[5]),
|
||||
.io_writeback_bits_data_6(writeback_data[6]),
|
||||
.io_writeback_bits_data_7(writeback_data[7]),
|
||||
.io_writeback_bits_data_0(tensor_writeback_data[0]),
|
||||
.io_writeback_bits_data_1(tensor_writeback_data[1]),
|
||||
.io_writeback_bits_data_2(tensor_writeback_data[2]),
|
||||
.io_writeback_bits_data_3(tensor_writeback_data[3]),
|
||||
.io_writeback_bits_data_4(tensor_writeback_data[4]),
|
||||
.io_writeback_bits_data_5(tensor_writeback_data[5]),
|
||||
.io_writeback_bits_data_6(tensor_writeback_data[6]),
|
||||
.io_writeback_bits_data_7(tensor_writeback_data[7]),
|
||||
|
||||
.io_respA_ready(tmem_if.rsp_ready),
|
||||
.io_respA_valid(tmem_if.rsp_valid),
|
||||
@@ -199,7 +201,9 @@ module VX_tensor_blackwell_core_block import VX_gpu_pkg::*; #(
|
||||
if (sync_launch_pending) begin
|
||||
if (sync_writeback_done) begin
|
||||
commit_select_tensor = 1'b1;
|
||||
metadata_deq = commit_if.ready;
|
||||
// BWGMMA retires when it launches. Its later tensor writeback
|
||||
// is a ghost commit and must not dequeue the next uop.
|
||||
metadata_deq = commit_if.ready && !sync_launch_is_bwgmma;
|
||||
commit_sync_done = commit_if.ready;
|
||||
end else if (sync_no_wb_done) begin
|
||||
commit_select_tensor = 1'b0;
|
||||
@@ -215,7 +219,7 @@ module VX_tensor_blackwell_core_block import VX_gpu_pkg::*; #(
|
||||
writeback_ready = 1'b0;
|
||||
commit_select_tensor = 1'b0;
|
||||
sync_done = initiate_valid && initiate_ready;
|
||||
metadata_deq = 1'b0;
|
||||
metadata_deq = sync_done;
|
||||
end else if (tcgen05_cp || tcgen05_ld || tcgen05_st || tcgen05_cb) begin
|
||||
writeback_ready = 1'b0;
|
||||
commit_select_tensor = 1'b0;
|
||||
@@ -245,6 +249,7 @@ module VX_tensor_blackwell_core_block import VX_gpu_pkg::*; #(
|
||||
commit_if.data.eop = sync_launch_is_ld ? 1'b1 : writeback_last;
|
||||
end else begin
|
||||
commit_if.valid = (sync_no_wb_done && metadata_valid) ||
|
||||
(metadata_valid && !sync_launch_pending && bwgmma && initiate_ready) ||
|
||||
(metadata_valid && !sync_launch_pending && !tensor_launch_op &&
|
||||
(!tensor_wait_op || tensor_idle));
|
||||
commit_if.data.uuid = execute_if_data_uuid;
|
||||
|
||||
Reference in New Issue
Block a user