Skip to content

Commit 38a5fe4

Browse files
author
Boyko Borisov
committed
Remove TriteiaLinear layer
1 parent b37d043 commit 38a5fe4

File tree

1 file changed

+0
-18
lines changed

1 file changed

+0
-18
lines changed

scratchpad/nn/layers/linear.py

Lines changed: 0 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -1162,21 +1162,3 @@ def extra_repr(self) -> str:
11621162
s += f", tp_size={self.tp_size}"
11631163
s += f", reduce_results={self.reduce_results}"
11641164
return s
1165-
1166-
class TritelaLinear(LinearBase):
1167-
def __init__(
1168-
self,
1169-
input_size,
1170-
output_size,
1171-
skip_bias_add = False,
1172-
params_dtype = None,
1173-
quant_config = None,
1174-
prefix = ""
1175-
):
1176-
super().__init__(
1177-
input_size, output_size, skip_bias_add, params_dtype, quant_config, prefix
1178-
)
1179-
self.layer = sparse_low_precision_linear(input_size, output_size)
1180-
1181-
def forward(self, x):
1182-
return self.layer(x)

0 commit comments

Comments
 (0)