Skip to content

Commit

Permalink
typo
Browse files Browse the repository at this point in the history
  • Loading branch information
liuanji committed Mar 10, 2024
1 parent ee0596b commit a419fdb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/pyjuice/layer/sum_layer.py
Original file line number Diff line number Diff line change
Expand Up @@ -2164,7 +2164,7 @@ def _backward_block_sparse_par_flows(self, node_flows: torch.Tensor, params: tor
TILE_SIZE_K = min(2048 // TILE_SIZE_B, num_edges)

if propagation_alg_id == 1:
# The kernel will stale if the tile sizes are too large
# The kernel will stall if the tile sizes are too large
TILE_SIZE_M = 16
TILE_SIZE_K = 16
TILE_SIZE_B = 16
Expand Down

0 comments on commit a419fdb

Please sign in to comment.