Skip to content

GitLab

Unverified Commit 4027bcba authored by Thor Johnsen's avatar Thor Johnsen Committed by GitHub
Browse files

Merge pull request #904 from ksivaman/sparsity 

Fixed variable name
parents 1ff54b8f 59995c76
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
apex/contrib/sparsity/asp.py
View file @ 4027bcba
...@@ -102,7 +102,7 @@ class ASP: ...@@ -102,7 +102,7 @@ class ASP:
print("[ASP] Sparsifying %s::%s of size=%s and type=%s for sparsity" % (module_name, p_name, str(p.size()), str(p.dtype))) print("[ASP] Sparsifying %s::%s of size=%s and type=%s for sparsity" % (module_name, p_name, str(p.size()), str(p.dtype)))
mask = torch.ones_like(p).bool() mask = torch.ones_like(p).bool()
buffname = name.split(".")[-1] # buffer names cannot contain "." buffname = p_name.split(".")[-1] # buffer names cannot contain "."
module.register_buffer('__%s_mma_mask' % buffname, mask) module.register_buffer('__%s_mma_mask' % buffname, mask)
if allow_recompute_mask: if allow_recompute_mask:
pruned = torch.zeros_like(p).cpu() pruned = torch.zeros_like(p).cpu()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment