I am very pleased to see that you have open-sourced such an outstanding work.
However, I encountered a problem while installing the training environment you provided. My graphics card is an RTX 5090, and when compiling your submodule gemm-gs, an error occurs. I looked at the setup.py file and changed the architecture to sm_120, but the error persists. May I ask if you have run your code on the 50 series graphics cards, and is there any solution to this problem? Will you maintain a version of the submodule for 50 series graphics cards in the future?
I am very pleased to see that you have open-sourced such an outstanding work.
However, I encountered a problem while installing the training environment you provided. My graphics card is an RTX 5090, and when compiling your submodule gemm-gs, an error occurs. I looked at the setup.py file and changed the architecture to sm_120, but the error persists. May I ask if you have run your code on the 50 series graphics cards, and is there any solution to this problem? Will you maintain a version of the submodule for 50 series graphics cards in the future?