Fix examples and pytest, run ruff (#3230)
Co-authored-by: dePaul Miller <23461061+depaulmillz@users.noreply.github.com>
This commit is contained in:
co-authored by
dePaul Miller
parent
982cb9e718
commit
546c3efa89
@@ -26,14 +26,14 @@
|
||||
# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
from blackwell.tutorial_gemm import fp16_gemm_0
|
||||
from blackwell.tutorial_gemm import fp16_gemm_1
|
||||
from blackwell.tutorial_gemm import fp16_gemm_2
|
||||
from blackwell.tutorial_gemm import fp16_gemm_3
|
||||
from blackwell.tutorial_gemm import fp16_gemm_3_1
|
||||
from blackwell.tutorial_gemm import fp16_gemm_4
|
||||
from blackwell.tutorial_gemm import fp16_gemm_5
|
||||
from blackwell.tutorial_gemm import fp16_gemm_6
|
||||
from blackwell.tutorial.tutorial_gemm import fp16_gemm_0
|
||||
from blackwell.tutorial.tutorial_gemm import fp16_gemm_1
|
||||
from blackwell.tutorial.tutorial_gemm import fp16_gemm_2
|
||||
from blackwell.tutorial.tutorial_gemm import fp16_gemm_3
|
||||
from blackwell.tutorial.tutorial_gemm import fp16_gemm_3_1
|
||||
from blackwell.tutorial.tutorial_gemm import fp16_gemm_4
|
||||
from blackwell.tutorial.tutorial_gemm import fp16_gemm_5
|
||||
from blackwell.tutorial.tutorial_gemm import fp16_gemm_6
|
||||
|
||||
import pytest
|
||||
from typing import Tuple
|
||||
@@ -63,7 +63,6 @@ def test_fp16_gemm_1(
|
||||
fp16_gemm_1.run_dense_gemm(mnk, tolerance)
|
||||
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"mnk",
|
||||
[(512, 512, 256)],
|
||||
|
||||
Reference in New Issue
Block a user