replicate
/

gpt-oss-triton-kernels

medmekk commited on Jan 27

Commit

68cce72

verified ·

1 Parent(s): 8cdbf3d

Update build/torch-rocm/init.py

Files changed (1) hide show

build/torch-rocm/__init__.py CHANGED Viewed

@@ -1,8 +1,8 @@
 # Make sure to add this in the build folder as this won't build if we put that here
-# from . import matmul_ogs, tensor_details, numerics_details, tensor, swiglu, routing
-# __all__ = ["matmul_ogs" , "tensor_details", "numerics_details", "tensor", "swiglu", "routing"]
 # Then, run the following code to build the kernels:
 # docker run --rm \

 # Make sure to add this in the build folder as this won't build if we put that here
+from . import matmul_ogs, tensor_details, numerics_details, tensor, swiglu, routing
+__all__ = ["matmul_ogs" , "tensor_details", "numerics_details", "tensor", "swiglu", "routing"]
 # Then, run the following code to build the kernels:
 # docker run --rm \