mirror of
https://github.com/ml-explore/mlx.git
synced 2025-10-19 08:29:35 +08:00
use fp32 for testing, add more complex ops (#2322)
This commit is contained in:
@@ -1,6 +1,10 @@
|
||||
# Copyright © 2023 Apple Inc.
|
||||
|
||||
import os
|
||||
|
||||
# Use regular fp32 precision for tests
|
||||
os.environ["MLX_ENABLE_TF32"] = "0"
|
||||
|
||||
import platform
|
||||
import unittest
|
||||
from typing import Any, Callable, List, Tuple, Union
|
||||
|
Reference in New Issue
Block a user